Skip to content

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

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…