use black format all files;

remove "return state" for functions which will be executed in vmap;
recover randkey as args in mutation methods

tensorneat/examples/brax/ant.py

@@ -16,7 +16,8 @@ if __name__ == '__main__':
                     node_gene=DefaultNodeGene(
                         activation_options=(Act.tanh,),
                         activation_default=Act.tanh,
-                    )
+                    ),
+                    output_transform=Act.tanh
                 ),
                 pop_size=1000,
                 species_size=10,

tensorneat/examples/with_evox/ray_test.py

@@ -0,0 +1,5 @@
+import ray
+ray.init(num_gpus=2)
+
+available_resources = ray.available_resources()
+print("Available resources:", available_resources)