diff --git a/neurax/__init__.py b/jaxley/__init__.py similarity index 100% rename from neurax/__init__.py rename to jaxley/__init__.py diff --git a/neurax/build_branched_tridiag.py b/jaxley/build_branched_tridiag.py similarity index 100% rename from neurax/build_branched_tridiag.py rename to jaxley/build_branched_tridiag.py diff --git a/neurax/channels/__init__.py b/jaxley/channels/__init__.py similarity index 100% rename from neurax/channels/__init__.py rename to jaxley/channels/__init__.py diff --git a/neurax/channels/channel.py b/jaxley/channels/channel.py similarity index 100% rename from neurax/channels/channel.py rename to jaxley/channels/channel.py diff --git a/neurax/channels/hh.py b/jaxley/channels/hh.py similarity index 100% rename from neurax/channels/hh.py rename to jaxley/channels/hh.py diff --git a/neurax/channels/pospischil.py b/jaxley/channels/pospischil.py similarity index 100% rename from neurax/channels/pospischil.py rename to jaxley/channels/pospischil.py diff --git a/neurax/connection.py b/jaxley/connection.py similarity index 100% rename from neurax/connection.py rename to jaxley/connection.py diff --git a/neurax/integrate.py b/jaxley/integrate.py similarity index 100% rename from neurax/integrate.py rename to jaxley/integrate.py diff --git a/neurax/modules/__init__.py b/jaxley/modules/__init__.py similarity index 100% rename from neurax/modules/__init__.py rename to jaxley/modules/__init__.py diff --git a/neurax/modules/base.py b/jaxley/modules/base.py similarity index 100% rename from neurax/modules/base.py rename to jaxley/modules/base.py diff --git a/neurax/modules/branch.py b/jaxley/modules/branch.py similarity index 100% rename from neurax/modules/branch.py rename to jaxley/modules/branch.py diff --git a/neurax/modules/cell.py b/jaxley/modules/cell.py similarity index 100% rename from neurax/modules/cell.py rename to jaxley/modules/cell.py diff --git a/neurax/modules/compartment.py b/jaxley/modules/compartment.py similarity index 100% rename from neurax/modules/compartment.py rename to jaxley/modules/compartment.py diff --git a/neurax/modules/network.py b/jaxley/modules/network.py similarity index 100% rename from neurax/modules/network.py rename to jaxley/modules/network.py diff --git a/neurax/optimize/__init__.py b/jaxley/optimize/__init__.py similarity index 100% rename from neurax/optimize/__init__.py rename to jaxley/optimize/__init__.py diff --git a/neurax/optimize/transforms.py b/jaxley/optimize/transforms.py similarity index 100% rename from neurax/optimize/transforms.py rename to jaxley/optimize/transforms.py diff --git a/neurax/solver_gate.py b/jaxley/solver_gate.py similarity index 100% rename from neurax/solver_gate.py rename to jaxley/solver_gate.py diff --git a/neurax/solver_voltage.py b/jaxley/solver_voltage.py similarity index 100% rename from neurax/solver_voltage.py rename to jaxley/solver_voltage.py diff --git a/neurax/stimulus.py b/jaxley/stimulus.py similarity index 100% rename from neurax/stimulus.py rename to jaxley/stimulus.py diff --git a/neurax/synapses/__init__.py b/jaxley/synapses/__init__.py similarity index 100% rename from neurax/synapses/__init__.py rename to jaxley/synapses/__init__.py diff --git a/neurax/synapses/glutamate.py b/jaxley/synapses/glutamate.py similarity index 100% rename from neurax/synapses/glutamate.py rename to jaxley/synapses/glutamate.py diff --git a/neurax/synapses/synapse.py b/jaxley/synapses/synapse.py similarity index 100% rename from neurax/synapses/synapse.py rename to jaxley/synapses/synapse.py diff --git a/neurax/utils/__init__.py b/jaxley/utils/__init__.py similarity index 100% rename from neurax/utils/__init__.py rename to jaxley/utils/__init__.py diff --git a/neurax/utils/cell_utils.py b/jaxley/utils/cell_utils.py similarity index 100% rename from neurax/utils/cell_utils.py rename to jaxley/utils/cell_utils.py diff --git a/neurax/utils/colors.py b/jaxley/utils/colors.py similarity index 100% rename from neurax/utils/colors.py rename to jaxley/utils/colors.py diff --git a/neurax/utils/jax_utils.py b/jaxley/utils/jax_utils.py similarity index 100% rename from neurax/utils/jax_utils.py rename to jaxley/utils/jax_utils.py diff --git a/neurax/utils/plot_utils.py b/jaxley/utils/plot_utils.py similarity index 100% rename from neurax/utils/plot_utils.py rename to jaxley/utils/plot_utils.py diff --git a/neurax/utils/swc.py b/jaxley/utils/swc.py similarity index 100% rename from neurax/utils/swc.py rename to jaxley/utils/swc.py diff --git a/neurax/utils/syn_utils.py b/jaxley/utils/syn_utils.py similarity index 100% rename from neurax/utils/syn_utils.py rename to jaxley/utils/syn_utils.py