diff --git a/.doctrees/Contribution.doctree b/.doctrees/Contribution.doctree
index 4b38ae83..0f35c30b 100644
Binary files a/.doctrees/Contribution.doctree and b/.doctrees/Contribution.doctree differ
diff --git a/.doctrees/Execution.doctree b/.doctrees/Execution.doctree
index 18fc6ea6..ac98ffaf 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 59c24c00..cc7dd3ea 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 41da68b0..4a18a743 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 bd7a4c02..83141c68 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 c589f9b5..c7e094e4 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 1fc0c739..3fb107ae 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 6c1f7495..8a4bd0d1 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 4a0fd801..b6b39388 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 a699fd1e..80b3d51c 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 cebfef62..0c59a607 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 23c5c5cb..aeb10075 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 8741923f..2cb479ff 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 d6eb68e3..cfe25965 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 c3e1867e..3d0b6808 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 ccd038f4..6d954ead 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 a457b32c..2981cab5 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 4d041605..ef9ec32f 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 7b25bb56..809e34ee 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 36b588af..fddbf658 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 41b1ac11..605a5b68 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 87bb21cf..bf8c89db 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 d18f45bd..9cddcd9e 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 1b8878e7..b2150a6a 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 286e18bc..5e6be569 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 a1263bcc..833d45f1 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 142f6ef7..aa8ef6fd 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 1dc9dd97..b1d45b69 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 ccfd5cf8..8dfdf7a4 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 053eb98c..d4c66b25 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 88d6305d..75f35ddd 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 25773ab3..f54efb50 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 fb7a5a05..282384e7 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 d349d3eb..88ec2578 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 b8b90f55..15716a41 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 0950f147..de96b4b8 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 892d3991..6a6770aa 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 09caaf45..a9e0d495 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 bd33ce45..f1dfa519 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 55f6f8b7..8a49f57e 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 8825498c..aa974350 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 4102374a..04d73104 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 64e0461e..f7a8374e 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 31de0669..6cf1da18 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 11790fdd..26d679b6 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 6a6c8ace..d0ee7398 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 91092423..8dfa72ed 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 e90e46d3..2b8f1390 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 29fba5be..f85f83db 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 2b0cda40..097e8a38 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 a371cfc9..79879fbc 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 24069158..4ffb0f91 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 59cedbe9..d8682a68 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 6de90561..15b7d546 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 84936651..1f33a2b7 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 d1294b4a..e598b5ba 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 d7dcc305..c1b8eaf1 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 82591765..a0763074 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 2d2a7e15..6b9d0e7e 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 d53e9b82..f1b3edb9 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 db9e5729..b8e2d3ae 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 4a01064c..574594ba 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 3361dfa2..500cf23f 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 3dca78b8..130483f4 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 152a174f..4b23d208 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 6bd46e8e..db30fa04 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 4cef9fff..ca82cdd8 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 a8199ed1..e22a94ae 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 50cf7b67..d82ca0f6 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 60f543ec..3b4a39cb 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 bbce165b..3bbe324f 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 2b7cfb62..25903dd7 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 64ae320d..69e08489 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 bf538d37..fc1c62f7 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 0f34cb0b..6ae7160c 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 fb15aa3c..0a689b5f 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 747d11ce..9de418e3 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 77dc102e..2a0b879e 100644
Binary files a/.doctrees/environment.pickle and b/.doctrees/environment.pickle differ
diff --git a/.doctrees/index.doctree b/.doctrees/index.doctree
index c87d1655..c0649be3 100644
Binary files a/.doctrees/index.doctree and b/.doctrees/index.doctree differ