change a lot a lot a lot!!!!!!!

examples/a.py

@@ -1,11 +1,28 @@
-import numpy as np
-import jax.numpy as jnp
+import jax
+from jax import numpy as jnp
 
-a = jnp.zeros((5, 5))
-k1 = jnp.array([1, 2, 3])
-k2 = jnp.array([2, 3, 4])
-v = jnp.array([1, 1, 1])
+from config import Config
+from core import Genome
 
-a = a.at[k1, k2].set(v)
+config = Config()
+from dataclasses import asdict
+
+print(asdict(config))
+
+pop_nodes = jnp.ones((Config.basic.pop_size, Config.neat.maximum_nodes, 3))
+pop_conns = jnp.ones((Config.basic.pop_size, Config.neat.maximum_conns, 5))
+
+pop_genomes1 = jax.vmap(Genome)(pop_nodes, pop_conns)
+pop_genomes2 = Genome(pop_nodes, pop_conns)
+
+print(pop_genomes)
+print(pop_genomes[0])
+
+@jax.vmap
+def pop_cnts(genome):
+    return genome.count()
+
+cnts = pop_cnts(pop_genomes)
+
+print(cnts)
 
-print(a)