Description:

Jaxley is a differentiable simulator for biophysical neuron models in JAX. Its key features are:

• automatic differentiation, allowing gradient-based optimization of thousands of parameters
• support for CPU, GPU, or TPU without any changes to the code
• jit-compilation, making it as fast as other packages while being fully written in python
• backward-Euler solver for stable numerical solution of multicompartment neurons
• elegant mechanisms for parameter sharing

License

For personal and professional use. You cannot resell or redistribute these repositories in their original state.

Share

• Share on Facebook
• Share on Twitter