diff --git a/tensorneat/algorithm/neat/neat.py b/tensorneat/algorithm/neat/neat.py
index 6c04d8e..32abd07 100644
--- a/tensorneat/algorithm/neat/neat.py
+++ b/tensorneat/algorithm/neat/neat.py
@@ -1,4 +1,3 @@
-import jax, jax.numpy as jnp
 from tensorneat.common import State
 from .. import BaseAlgorithm
 from .species import *
diff --git a/tensorneat/algorithm/neat/genome/__init__.py b/tensorneat/genome/__init__.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/__init__.py
rename to tensorneat/genome/__init__.py
diff --git a/tensorneat/algorithm/neat/genome/base.py b/tensorneat/genome/base.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/base.py
rename to tensorneat/genome/base.py
diff --git a/tensorneat/algorithm/neat/genome/default.py b/tensorneat/genome/default.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/default.py
rename to tensorneat/genome/default.py
diff --git a/tensorneat/algorithm/neat/gene/__init__.py b/tensorneat/genome/gene/__init__.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/__init__.py
rename to tensorneat/genome/gene/__init__.py
diff --git a/tensorneat/algorithm/neat/gene/base.py b/tensorneat/genome/gene/base.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/base.py
rename to tensorneat/genome/gene/base.py
diff --git a/tensorneat/algorithm/neat/gene/conn/__init__.py b/tensorneat/genome/gene/conn/__init__.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/conn/__init__.py
rename to tensorneat/genome/gene/conn/__init__.py
diff --git a/tensorneat/algorithm/neat/gene/conn/base.py b/tensorneat/genome/gene/conn/base.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/conn/base.py
rename to tensorneat/genome/gene/conn/base.py
diff --git a/tensorneat/algorithm/neat/gene/conn/default.py b/tensorneat/genome/gene/conn/default.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/conn/default.py
rename to tensorneat/genome/gene/conn/default.py
diff --git a/tensorneat/algorithm/neat/gene/node/__init__.py b/tensorneat/genome/gene/node/__init__.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/node/__init__.py
rename to tensorneat/genome/gene/node/__init__.py
diff --git a/tensorneat/algorithm/neat/gene/node/base.py b/tensorneat/genome/gene/node/base.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/node/base.py
rename to tensorneat/genome/gene/node/base.py
diff --git a/tensorneat/algorithm/neat/gene/node/bias.py b/tensorneat/genome/gene/node/bias.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/node/bias.py
rename to tensorneat/genome/gene/node/bias.py
diff --git a/tensorneat/algorithm/neat/gene/node/default.py b/tensorneat/genome/gene/node/default.py
similarity index 100%
rename from tensorneat/algorithm/neat/gene/node/default.py
rename to tensorneat/genome/gene/node/default.py
diff --git a/tensorneat/algorithm/neat/genome/operations/__init__.py b/tensorneat/genome/operations/__init__.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/__init__.py
rename to tensorneat/genome/operations/__init__.py
diff --git a/tensorneat/algorithm/neat/genome/operations/crossover/__init__.py b/tensorneat/genome/operations/crossover/__init__.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/crossover/__init__.py
rename to tensorneat/genome/operations/crossover/__init__.py
diff --git a/tensorneat/algorithm/neat/genome/operations/crossover/base.py b/tensorneat/genome/operations/crossover/base.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/crossover/base.py
rename to tensorneat/genome/operations/crossover/base.py
diff --git a/tensorneat/algorithm/neat/genome/operations/crossover/default.py b/tensorneat/genome/operations/crossover/default.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/crossover/default.py
rename to tensorneat/genome/operations/crossover/default.py
diff --git a/tensorneat/algorithm/neat/genome/operations/distance/__init__.py b/tensorneat/genome/operations/distance/__init__.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/distance/__init__.py
rename to tensorneat/genome/operations/distance/__init__.py
diff --git a/tensorneat/algorithm/neat/genome/operations/distance/base.py b/tensorneat/genome/operations/distance/base.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/distance/base.py
rename to tensorneat/genome/operations/distance/base.py
diff --git a/tensorneat/algorithm/neat/genome/operations/distance/default.py b/tensorneat/genome/operations/distance/default.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/distance/default.py
rename to tensorneat/genome/operations/distance/default.py
diff --git a/tensorneat/algorithm/neat/genome/operations/mutation/__init__.py b/tensorneat/genome/operations/mutation/__init__.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/mutation/__init__.py
rename to tensorneat/genome/operations/mutation/__init__.py
diff --git a/tensorneat/algorithm/neat/genome/operations/mutation/base.py b/tensorneat/genome/operations/mutation/base.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/mutation/base.py
rename to tensorneat/genome/operations/mutation/base.py
diff --git a/tensorneat/algorithm/neat/genome/operations/mutation/default.py b/tensorneat/genome/operations/mutation/default.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/operations/mutation/default.py
rename to tensorneat/genome/operations/mutation/default.py
diff --git a/tensorneat/algorithm/neat/genome/recurrent.py b/tensorneat/genome/recurrent.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/recurrent.py
rename to tensorneat/genome/recurrent.py
diff --git a/tensorneat/algorithm/neat/genome/utils.py b/tensorneat/genome/utils.py
similarity index 100%
rename from tensorneat/algorithm/neat/genome/utils.py
rename to tensorneat/genome/utils.py