diff --git a/.doctrees/Contribution.doctree b/.doctrees/Contribution.doctree
index 0f35c30b..a1d079fe 100644
Binary files a/.doctrees/Contribution.doctree and b/.doctrees/Contribution.doctree differ
diff --git a/.doctrees/Execution.doctree b/.doctrees/Execution.doctree
index ac98ffaf..2a50cffd 100644
Binary files a/.doctrees/Execution.doctree and b/.doctrees/Execution.doctree differ
diff --git a/.doctrees/Execution_On_a_Slurm_cluster.doctree b/.doctrees/Execution_On_a_Slurm_cluster.doctree
index cc7dd3ea..5897d296 100644
Binary files a/.doctrees/Execution_On_a_Slurm_cluster.doctree and b/.doctrees/Execution_On_a_Slurm_cluster.doctree differ
diff --git a/.doctrees/Execution_On_an_Ubuntu_machine.doctree b/.doctrees/Execution_On_an_Ubuntu_machine.doctree
index 4a18a743..9494d122 100644
Binary files a/.doctrees/Execution_On_an_Ubuntu_machine.doctree and b/.doctrees/Execution_On_an_Ubuntu_machine.doctree differ
diff --git a/.doctrees/Installation.doctree b/.doctrees/Installation.doctree
index 83141c68..2cd7923c 100644
Binary files a/.doctrees/Installation.doctree and b/.doctrees/Installation.doctree differ
diff --git a/.doctrees/Installation_Common_to_all_machines.doctree b/.doctrees/Installation_Common_to_all_machines.doctree
index c7e094e4..d6148895 100644
Binary files a/.doctrees/Installation_Common_to_all_machines.doctree and b/.doctrees/Installation_Common_to_all_machines.doctree differ
diff --git a/.doctrees/Installation_On_a_Slurm_cluster.doctree b/.doctrees/Installation_On_a_Slurm_cluster.doctree
index 3fb107ae..1a2d7dc5 100644
Binary files a/.doctrees/Installation_On_a_Slurm_cluster.doctree and b/.doctrees/Installation_On_a_Slurm_cluster.doctree differ
diff --git a/.doctrees/Installation_On_an_Ubuntu_machine.doctree b/.doctrees/Installation_On_an_Ubuntu_machine.doctree
index 8a4bd0d1..1224e60e 100644
Binary files a/.doctrees/Installation_On_an_Ubuntu_machine.doctree and b/.doctrees/Installation_On_an_Ubuntu_machine.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.config.doctree b/.doctrees/_autosummary/cneuromax.config.doctree
index b6b39388..731ed611 100644
Binary files a/.doctrees/_autosummary/cneuromax.config.doctree and b/.doctrees/_autosummary/cneuromax.config.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.doctree b/.doctrees/_autosummary/cneuromax.doctree
index 80b3d51c..8c2e104f 100644
Binary files a/.doctrees/_autosummary/cneuromax.doctree and b/.doctrees/_autosummary/cneuromax.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.config.doctree b/.doctrees/_autosummary/cneuromax.fitting.config.doctree
index 0c59a607..aa5b9fb6 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.config.doctree and b/.doctrees/_autosummary/cneuromax.fitting.config.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.config.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.config.doctree
index aeb10075..421ebf37 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.config.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.config.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.datamodule.base.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.datamodule.base.doctree
index 2cb479ff..497a98db 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.datamodule.base.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.datamodule.base.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.datamodule.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.datamodule.doctree
index cfe25965..8cb80f72 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.datamodule.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.datamodule.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.doctree
index 3d0b6808..9180f44b 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.base.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.base.doctree
index 6d954ead..ca12a4b7 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.base.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.base.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.classification.base.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.classification.base.doctree
index 2981cab5..1f9dace0 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.classification.base.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.classification.base.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.classification.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.classification.doctree
index ef9ec32f..f8f1a009 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.classification.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.classification.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.doctree
index 809e34ee..b574ac49 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.nnmodule.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.nnmodule.doctree
index fddbf658..acfd3d9e 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.nnmodule.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.nnmodule.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.nnmodule.mlp.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.nnmodule.mlp.doctree
index 605a5b68..15bac650 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.nnmodule.mlp.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.nnmodule.mlp.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.store.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.store.doctree
index bf8c89db..1d510c6c 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.store.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.store.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.wandb_val_logging.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.wandb_val_logging.doctree
index 9cddcd9e..43aeefbe 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.wandb_val_logging.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.litmodule.wandb_val_logging.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.runner.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.runner.doctree
index b2150a6a..a3735881 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.runner.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.runner.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.store.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.store.doctree
index 5e6be569..53abf2b0 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.store.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.store.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.train.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.train.doctree
index 833d45f1..0ad59efe 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.train.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.train.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.doctree
index aa8ef6fd..3aebc6b6 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.lightning.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.lightning.doctree
index b1d45b69..83f2550d 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.lightning.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.lightning.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplneuroevo.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplneuroevo.doctree
index 8dfdf7a4..f4f0aece 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplneuroevo.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplneuroevo.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.doctree b/.doctrees/_autosummary/cneuromax.fitting.doctree
index d4c66b25..30fd7c4a 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.doctree and b/.doctrees/_autosummary/cneuromax.fitting.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.agent.base.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.agent.base.doctree
index 75f35ddd..bc6b30a2 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.agent.base.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.agent.base.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.agent.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.agent.doctree
index f54efb50..0cefd86c 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.agent.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.agent.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.config.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.config.doctree
index 282384e7..6063aa1b 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.config.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.config.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.doctree
index 88ec2578..ad838107 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.fit.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.fit.doctree
index 15716a41..df614406 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.fit.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.fit.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.doctree
index de96b4b8..bbfcb61d 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.static.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.static.doctree
index 6a6770aa..8f85e1b3 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.static.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.static.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.static.rnnfc.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.static.rnnfc.doctree
index a9e0d495..51e9251b 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.static.rnnfc.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.cpu.static.rnnfc.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.doctree
index f1dfa519..cdcab1e8 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.net.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.runner.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.runner.doctree
index 8a49f57e..b5797886 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.runner.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.runner.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.base.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.base.doctree
index aa974350..a2c2e626 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.base.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.base.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.doctree
index 04d73104..68b07522 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.reinforcement.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.reinforcement.doctree
index f7a8374e..a30d03c6 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.reinforcement.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.space.reinforcement.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.compute.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.compute.doctree
index 6cf1da18..4aaa83ef 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.compute.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.compute.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.doctree
index 26d679b6..23bdce34 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.evolve.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.evolve.doctree
index d0ee7398..3ff5b403 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.evolve.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.evolve.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.exchange.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.exchange.doctree
index 8dfa72ed..92e29bfd 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.exchange.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.exchange.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.initialize.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.initialize.doctree
index 2b8f1390..ebedd5d3 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.initialize.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.initialize.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.readwrite.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.readwrite.doctree
index f85f83db..59f01986 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.readwrite.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.readwrite.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.type.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.type.doctree
index 097e8a38..0376af63 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.type.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.type.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.validate.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.validate.doctree
index 79879fbc..f0a07f3d 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.validate.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.validate.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.wandb.doctree b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.wandb.doctree
index 4ffb0f91..b4697821 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.wandb.doctree and b/.doctrees/_autosummary/cneuromax.fitting.neuroevolution.utils.wandb.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.runner.doctree b/.doctrees/_autosummary/cneuromax.fitting.runner.doctree
index d8682a68..537243d5 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.runner.doctree and b/.doctrees/_autosummary/cneuromax.fitting.runner.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.store.doctree b/.doctrees/_autosummary/cneuromax.fitting.store.doctree
index 15b7d546..4c2f4b09 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.store.doctree and b/.doctrees/_autosummary/cneuromax.fitting.store.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.utils.doctree b/.doctrees/_autosummary/cneuromax.fitting.utils.doctree
index 1f33a2b7..a6780dd2 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.utils.doctree and b/.doctrees/_autosummary/cneuromax.fitting.utils.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.utils.hydra.doctree b/.doctrees/_autosummary/cneuromax.fitting.utils.hydra.doctree
index e598b5ba..c04c1975 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.utils.hydra.doctree and b/.doctrees/_autosummary/cneuromax.fitting.utils.hydra.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.projects.classify_mnist.datamodule.doctree b/.doctrees/_autosummary/cneuromax.projects.classify_mnist.datamodule.doctree
index c1b8eaf1..1a3c7f36 100644
Binary files a/.doctrees/_autosummary/cneuromax.projects.classify_mnist.datamodule.doctree and b/.doctrees/_autosummary/cneuromax.projects.classify_mnist.datamodule.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.projects.classify_mnist.datamodule_test.doctree b/.doctrees/_autosummary/cneuromax.projects.classify_mnist.datamodule_test.doctree
index a0763074..e436cd62 100644
Binary files a/.doctrees/_autosummary/cneuromax.projects.classify_mnist.datamodule_test.doctree and b/.doctrees/_autosummary/cneuromax.projects.classify_mnist.datamodule_test.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.projects.classify_mnist.doctree b/.doctrees/_autosummary/cneuromax.projects.classify_mnist.doctree
index 6b9d0e7e..5e153050 100644
Binary files a/.doctrees/_autosummary/cneuromax.projects.classify_mnist.doctree and b/.doctrees/_autosummary/cneuromax.projects.classify_mnist.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.projects.classify_mnist.litmodule.doctree b/.doctrees/_autosummary/cneuromax.projects.classify_mnist.litmodule.doctree
index f1b3edb9..7fdce82f 100644
Binary files a/.doctrees/_autosummary/cneuromax.projects.classify_mnist.litmodule.doctree and b/.doctrees/_autosummary/cneuromax.projects.classify_mnist.litmodule.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.projects.doctree b/.doctrees/_autosummary/cneuromax.projects.doctree
index b8e2d3ae..2b74d522 100644
Binary files a/.doctrees/_autosummary/cneuromax.projects.doctree and b/.doctrees/_autosummary/cneuromax.projects.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.agent.doctree b/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.agent.doctree
index 574594ba..dbc0a94c 100644
Binary files a/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.agent.doctree and b/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.agent.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.doctree b/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.doctree
index 500cf23f..6da90bb6 100644
Binary files a/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.doctree and b/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.space.doctree b/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.space.doctree
index 130483f4..9fc4199d 100644
Binary files a/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.space.doctree and b/.doctrees/_autosummary/cneuromax.projects.neuroevorl_control.space.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.runner.doctree b/.doctrees/_autosummary/cneuromax.runner.doctree
index 4b23d208..157a63df 100644
Binary files a/.doctrees/_autosummary/cneuromax.runner.doctree and b/.doctrees/_autosummary/cneuromax.runner.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.serving.doctree b/.doctrees/_autosummary/cneuromax.serving.doctree
index db30fa04..bc186667 100644
Binary files a/.doctrees/_autosummary/cneuromax.serving.doctree and b/.doctrees/_autosummary/cneuromax.serving.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.store.doctree b/.doctrees/_autosummary/cneuromax.store.doctree
index ca82cdd8..7215e7ae 100644
Binary files a/.doctrees/_autosummary/cneuromax.store.doctree and b/.doctrees/_autosummary/cneuromax.store.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.testing.doctree b/.doctrees/_autosummary/cneuromax.testing.doctree
index e22a94ae..1acd924b 100644
Binary files a/.doctrees/_autosummary/cneuromax.testing.doctree and b/.doctrees/_autosummary/cneuromax.testing.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.beartype.doctree b/.doctrees/_autosummary/cneuromax.utils.beartype.doctree
index d82ca0f6..733b1366 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.beartype.doctree and b/.doctrees/_autosummary/cneuromax.utils.beartype.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.doctree b/.doctrees/_autosummary/cneuromax.utils.doctree
index 3b4a39cb..88a7841f 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.doctree and b/.doctrees/_autosummary/cneuromax.utils.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.gymnasium_test.doctree b/.doctrees/_autosummary/cneuromax.utils.gymnasium_test.doctree
index 3bbe324f..5407d839 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.gymnasium_test.doctree and b/.doctrees/_autosummary/cneuromax.utils.gymnasium_test.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.hydra_zen.doctree b/.doctrees/_autosummary/cneuromax.utils.hydra_zen.doctree
index 25903dd7..fd6a8448 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.hydra_zen.doctree and b/.doctrees/_autosummary/cneuromax.utils.hydra_zen.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.misc.doctree b/.doctrees/_autosummary/cneuromax.utils.misc.doctree
index 69e08489..10b272b5 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.misc.doctree and b/.doctrees/_autosummary/cneuromax.utils.misc.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.mpi4py.doctree b/.doctrees/_autosummary/cneuromax.utils.mpi4py.doctree
index fc1c62f7..9d22b53a 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.mpi4py.doctree and b/.doctrees/_autosummary/cneuromax.utils.mpi4py.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.runner.doctree b/.doctrees/_autosummary/cneuromax.utils.runner.doctree
index 6ae7160c..24eb2f98 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.runner.doctree and b/.doctrees/_autosummary/cneuromax.utils.runner.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.torch.doctree b/.doctrees/_autosummary/cneuromax.utils.torch.doctree
index 0a689b5f..1b9722fa 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.torch.doctree and b/.doctrees/_autosummary/cneuromax.utils.torch.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.utils.wandb.doctree b/.doctrees/_autosummary/cneuromax.utils.wandb.doctree
index 9de418e3..1718c09a 100644
Binary files a/.doctrees/_autosummary/cneuromax.utils.wandb.doctree and b/.doctrees/_autosummary/cneuromax.utils.wandb.doctree differ
diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
index a5badc4e..5fe62431 100644
Binary files a/.doctrees/environment.pickle and b/.doctrees/environment.pickle differ
diff --git a/.doctrees/index.doctree b/.doctrees/index.doctree
index d21fc8cf..5ec5ed92 100644
Binary files a/.doctrees/index.doctree and b/.doctrees/index.doctree differ