use jit().lower.compile in create functions

2023-05-08 02:35:04 +08:00
parent 497d89fc69
commit d4a75b9394
9 changed files with 120 additions and 77 deletions
--- a/algorithms/neat/species.py
+++ b/algorithms/neat/species.py
@@ -5,6 +5,7 @@ import jax
 import numpy as np
 from numpy.typing import NDArray

+
 class Species(object):

    def __init__(self, key, generation):