Perfect!

Next is to connect with Evox!

examples/jax_playground.py

@@ -1,20 +1,16 @@
-from functools import partial
-
 import numpy as np
-import jax
 from jax import jit
 
 from configs import Configer
 from neat.pipeline import Pipeline
-from neat.function_factory import FunctionFactory
 
 xor_inputs = np.array([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=np.float32)
 xor_outputs = np.array([[0], [1], [1], [0]], dtype=np.float32)
 
 def main():
     config = Configer.load_config("xor.ini")
-    function_factory = FunctionFactory(config)
-    pipeline = Pipeline(config, function_factory)
+    print(config)
+    pipeline = Pipeline(config)
     forward_func = pipeline.ask()
     # inputs = np.tile(xor_inputs, (150, 1, 1))
     outputs = forward_func(xor_inputs)