diff --git a/.doctrees/Contribution.doctree b/.doctrees/Contribution.doctree
index 0f35c30b..b82f8c46 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..46f00c10 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..16bc9785 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..8d0d4948 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..aebf0d33 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..f803a763 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..c16c9ab3 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..5b319255 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..02e794c1 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..a40c3c5a 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..d2cf80ac 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 65ee38af..61e42775 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 8d2d999d..884783a9 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..bb82b43a 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..aa7d6b53 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 4f917d94..16e50006 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..4b2ea6a8 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..a140d981 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..646c4c81 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..3e9da26f 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..7540c1f7 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..65e95c66 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..3b6bede6 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..2697da2c 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..dde1d325 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..2c9ab0cd 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 d5d4f9a6..4aa24311 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..a40b4a65 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.deeplearning.utils.typing.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.typing.doctree
index da012b67..6bf1ade5 100644
Binary files a/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.typing.doctree and b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.utils.typing.doctree differ
diff --git a/.doctrees/_autosummary/cneuromax.fitting.deeplneuroevo.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplneuroevo.doctree
index 8dfdf7a4..97d0fc87 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..3ba7505d 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..65a607f9 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..7b571eb5 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..b8190d12 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..d42b7db0 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..b17818ff 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..96ad2f7b 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..ddde24d4 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..2a216e26 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..04cc99dd 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..0d1ce2e7 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..3719d0fd 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..00d224c4 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..3696a36d 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..63eed75b 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..75ee1afa 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..6e160816 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..0d27e5f3 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..b5f10b03 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..48289674 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..a10506f4 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..493a0909 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..5643e82a 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..bf17ed9b 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..153f561a 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..bb508c66 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..24f1c642 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 b59f6568..344cb1b6 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..0f0bb2ab 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..6fc74c65 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..305a628d 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..cc5bc0c4 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..86b2096b 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..f088f6b5 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..5f98fcf6 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..aff59648 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..c17089a4 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..c1146278 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..d0b9cabd 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..86d838d5 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..226a092e 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..623d3fee 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..eb4b3dd3 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..fe30f490 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..543d0e70 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..aa3d09f1 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..3a5d1c8c 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..632cdd31 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 8958eb34..ddd09bd9 100644
Binary files a/.doctrees/environment.pickle and b/.doctrees/environment.pickle differ
diff --git a/.doctrees/index.doctree b/.doctrees/index.doctree
index daeaccb7..d7af75f9 100644
Binary files a/.doctrees/index.doctree and b/.doctrees/index.doctree differ