What kinds of models can be implemented in Jaxley?#
Jaxley focuses on biophysical, Hodgkin-Huxley-type models. You can think of Jaxley like the NEURON simulator written in JAX.
Jaxley allows to simulate the following types of models, as well as networks thereof:
single-compartment (point neuron) Hodgkin-Huxley models
multi-compartment Hodgkin-Huxley models
rate-based neuron models (tutorial here)
For all of these models, Jaxley is flexible and accurate. For example, it can flexibly add new channel models, use different kinds of synapses (conductance-based, tanh, …), and it can insert different kinds of channels in different branches (or compartments) within single cells. Like NEURON, Jaxley implements a backward-Euler solver for stable numerical solution of multi-compartment neurons.
However, Jaxley does not implement the following types of models:
leaky-integrate and fire neurons
Ishikevich neuron models
etc…