feedbax

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Brian is a free, open source simulator for spiking neural networks.

Reinforcement learning environments with musculoskeletal models

A list of schools and researchers in computational neuroscience

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

Easy whole-brain modeling for computational neuroscientists ??????

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

Multiple dispatch over abstract array types in JAX.

A Python package of computer vision models for the Equinox ecosystem.