diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
index b92d6c45..888cc42b 100644
Binary files a/.doctrees/environment.pickle and b/.doctrees/environment.pickle differ
diff --git a/.doctrees/index.doctree b/.doctrees/index.doctree
index 2ae4737e..6aa773b0 100644
Binary files a/.doctrees/index.doctree and b/.doctrees/index.doctree differ
diff --git a/_modules/cneuromax/fitting/deeplearning/litmodule/store.html b/_modules/cneuromax/fitting/deeplearning/litmodule/store.html
index 7209953f..6fec93b6 100644
--- a/_modules/cneuromax/fitting/deeplearning/litmodule/store.html
+++ b/_modules/cneuromax/fitting/deeplearning/litmodule/store.html
@@ -353,6 +353,7 @@ <h1>Source code for cneuromax.fitting.deeplearning.litmodule.store</h1><div clas
 <span></span><span class="sd">&quot;&quot;&quot;:class:`.BaseLitModule` `Hydra &lt;https://hydra.cc&gt;`_ config store.&quot;&quot;&quot;</span>
 
 <span class="kn">from</span> <span class="nn">hydra_zen</span> <span class="kn">import</span> <span class="n">ZenStore</span>
+<span class="kn">from</span> <span class="nn">schedulefree</span> <span class="kn">import</span> <span class="n">AdamWScheduleFree</span>
 <span class="kn">from</span> <span class="nn">torch.optim</span> <span class="kn">import</span> <span class="n">SGD</span><span class="p">,</span> <span class="n">Adam</span><span class="p">,</span> <span class="n">AdamW</span>
 <span class="kn">from</span> <span class="nn">transformers</span> <span class="kn">import</span> <span class="p">(</span>
     <span class="n">get_constant_schedule</span><span class="p">,</span>
@@ -403,6 +404,11 @@ <h1>Source code for cneuromax.fitting.deeplearning.litmodule.store</h1><div clas
 <span class="sd">    &quot;&quot;&quot;</span>
     <span class="n">store</span><span class="p">(</span><span class="n">pfs_builds</span><span class="p">(</span><span class="n">Adam</span><span class="p">),</span> <span class="n">name</span><span class="o">=</span><span class="s2">&quot;adam&quot;</span><span class="p">,</span> <span class="n">group</span><span class="o">=</span><span class="s2">&quot;litmodule/optimizer&quot;</span><span class="p">)</span>
     <span class="n">store</span><span class="p">(</span><span class="n">pfs_builds</span><span class="p">(</span><span class="n">AdamW</span><span class="p">),</span> <span class="n">name</span><span class="o">=</span><span class="s2">&quot;adamw&quot;</span><span class="p">,</span> <span class="n">group</span><span class="o">=</span><span class="s2">&quot;litmodule/optimizer&quot;</span><span class="p">)</span>
+    <span class="n">store</span><span class="p">(</span>
+        <span class="n">pfs_builds</span><span class="p">(</span><span class="n">AdamWScheduleFree</span><span class="p">),</span>
+        <span class="n">name</span><span class="o">=</span><span class="s2">&quot;sfadamw&quot;</span><span class="p">,</span>
+        <span class="n">group</span><span class="o">=</span><span class="s2">&quot;litmodule/optimizer&quot;</span><span class="p">,</span>
+    <span class="p">)</span>
     <span class="n">store</span><span class="p">(</span><span class="n">pfs_builds</span><span class="p">(</span><span class="n">SGD</span><span class="p">),</span> <span class="n">name</span><span class="o">=</span><span class="s2">&quot;sgd&quot;</span><span class="p">,</span> <span class="n">group</span><span class="o">=</span><span class="s2">&quot;litmodule/optimizer&quot;</span><span class="p">)</span></div>