jaxley
Last updated:
0 purchases
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.
Customer Reviews
There are no reviews.
