diff --git a/cfg/config.yaml b/cfg/config.yaml
index 33e8193..0694af8 100644
--- a/cfg/config.yaml
+++ b/cfg/config.yaml
@@ -1,3 +1,4 @@
+# @package _global_
 defaults:
   # - model: lstm
   - dataset: mnist
diff --git a/cfg/dataset/adding.yaml b/cfg/dataset/adding.yaml
index 81b30e5..c59db8c 100644
--- a/cfg/dataset/adding.yaml
+++ b/cfg/dataset/adding.yaml
@@ -1,5 +1,5 @@
-dataset:
-  name: adding
-  L: 1000
-  samples: 55000
-  test_samples: 5000
+# @package _group_
+name: adding
+L: 1000
+samples: 55000
+test_samples: 5000
diff --git a/cfg/dataset/copying.yaml b/cfg/dataset/copying.yaml
index 42404f0..55ff203 100644
--- a/cfg/dataset/copying.yaml
+++ b/cfg/dataset/copying.yaml
@@ -1,9 +1,9 @@
+# @package _group_
 # Copying task
-dataset:
-  name: copying
-  L: 100 # length
-  M: 10  # number of tokens to memorize
-  A: 10  # alphabet size
-  variable: False # Randomly distribute memorization tokens throughout sequence instead of frontloading them
-  samples: 55000
-  test_samples: 5000
+name: copying
+L: 100 # length
+M: 10  # number of tokens to memorize
+A: 10  # alphabet size
+variable: False # Randomly distribute memorization tokens throughout sequence instead of frontloading them
+samples: 55000
+test_samples: 5000
diff --git a/cfg/dataset/ct.yaml b/cfg/dataset/ct.yaml
index 55a81e8..c5202a9 100644
--- a/cfg/dataset/ct.yaml
+++ b/cfg/dataset/ct.yaml
@@ -1,14 +1,14 @@
+# @package _group_
 # CharacterTrajectories
-dataset:
-  name: ct
-  # Sampling frequency relative to original sequence
-  train_hz: 1
-  eval_hz: 1
-  # Sample evenly or randomly
-  train_uniform: True
-  eval_uniform: True
-  # Include timestamps in input
-  timestamp: False
-  # Timestamp scale (multiplier on timestamp)
-  train_ts: 1
-  eval_ts: 1
+name: ct
+# Sampling frequency relative to original sequence
+train_hz: 1
+eval_hz: 1
+# Sample evenly or randomly
+train_uniform: True
+eval_uniform: True
+# Include timestamps in input
+timestamp: False
+# Timestamp scale (multiplier on timestamp)
+train_ts: 1
+eval_ts: 1
diff --git a/cfg/dataset/imdb.yaml b/cfg/dataset/imdb.yaml
index efdca14..2eda47f 100644
--- a/cfg/dataset/imdb.yaml
+++ b/cfg/dataset/imdb.yaml
@@ -1,3 +1,4 @@
+# @package _global_
 dataset:
   name: imdb
   vocab_size: 25000
diff --git a/cfg/dataset/mnist.yaml b/cfg/dataset/mnist.yaml
index 96147ab..25dc622 100644
--- a/cfg/dataset/mnist.yaml
+++ b/cfg/dataset/mnist.yaml
@@ -1,3 +1,3 @@
-dataset:
-  name: mnist
-  permute: False
+# @package _group_
+name: mnist
+permute: False
diff --git a/cfg/runner/pl.yaml b/cfg/runner/pl.yaml
index 15120ad..65f9296 100644
--- a/cfg/runner/pl.yaml
+++ b/cfg/runner/pl.yaml
@@ -1,3 +1,4 @@
+# @package _global_
 runner:
   name: pl
   ntrials: 1