diff --git a/.doctrees/Contribution.doctree b/.doctrees/Contribution.doctree
index 0f35c30b..356bd9eb 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..f429371d 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..02733739 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..36985ba6 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..b0cf7c7d 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..fbe5174b 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..cd9b9932 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..cd52f399 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..92946139 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..277cc527 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..751f9d52 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..e8a3f971 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 f1380b3e..097d6a91 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..e9e90f95 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..6a83dd7d 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 958a4587..7276ce59 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 297e4591..63fd88f7 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..976db1fe 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 ca5542ac..5d50df24 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..64e6809c 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..c53fe736 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..af192f54 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.runner.doctree b/.doctrees/_autosummary/cneuromax.fitting.deeplearning.runner.doctree
index b2150a6a..23d6d20f 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..f5637e1c 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..68e7e0bd 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..d0f75f52 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..e5165860 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..aa9dc85c 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..2ba7b8ea 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..110aa0ab 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..d008e994 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..81cd8a4b 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..e97f335a 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..dab465bc 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..b8c626e5 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..d636a63b 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..b9b4d0ac 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..a00f0ac9 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..57362b01 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..128f3815 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..b125f101 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..315bdfab 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..618a1d0a 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..32af3dbe 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..7c2e8f3c 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..e0b58746 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..63e9b326 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..df009c85 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..1faea2e7 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..48eeb7c7 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..3f9cc2d8 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..05266ee6 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..e08c5ead 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..a58335aa 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..3383e36e 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 0c804c54..d0819777 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..0c260402 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..471a7ac1 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 77027c44..3d5702d5 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..ffa4838a 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..47b2e8ca 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..105671c7 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..1600d3a4 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..8b59f280 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..110df180 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..94c77b03 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..174d322d 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..d9c65905 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..13ccbc3c 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..ae5ea2bc 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..93bece87 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..ee1a041a 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..18e9add2 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..636ba62b 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..70820a54 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..8ca6c3b0 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 67d0bbbc..6c20a705 100644
Binary files a/.doctrees/environment.pickle and b/.doctrees/environment.pickle differ
diff --git a/.doctrees/index.doctree b/.doctrees/index.doctree
index 6947fc5e..182eb05b 100644
Binary files a/.doctrees/index.doctree and b/.doctrees/index.doctree differ