PGI15 / snnaxView external linksLinks
Spiking neural networks in JAX with equinox.
☆28Aug 29, 2024Updated last year
Alternatives and similar repositories for snnax
Users that are interested in snnax are comparing it to the libraries listed below
Sorting:
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 5 months ago
- Code to accompany our paper "The combination of Hebbian and predictive plasticity learns invariant object representations in deep sensory…☆31Jan 14, 2025Updated last year
- jaxsnn is an event-based approach to machine-learning-inspired training and simulation of SNNs, including support for the BrainScaleS-2 n…☆30Feb 9, 2026Updated last week
- Coordinates in JAX☆29Jan 31, 2026Updated 2 weeks ago
- A differentiable spiking neural network simulator☆24Dec 5, 2025Updated 2 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆56Dec 4, 2024Updated last year
- ☆102Mar 18, 2024Updated last year
- Orthogonal polynomials with JAX☆27Jan 20, 2026Updated 3 weeks ago
- Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.☆19Jun 11, 2023Updated 2 years ago
- Parallel hyperparameter tuning with JAX☆39Jul 21, 2025Updated 6 months ago
- Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.☆178Updated this week
- A library for coreset algorithms, written in JAX for fast execution and GPU support.☆37Updated this week
- A library for deep learning with Spiking Neural Networks (SNN).☆38Updated this week
- ☆41Apr 27, 2022Updated 3 years ago
- Introduction to neuroimaging machine learning tool Nilearn☆12Oct 24, 2024Updated last year
- ☆11Feb 9, 2026Updated last week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Jan 16, 2026Updated last month
- A minimal JAX library for connectivity modelling at scale☆11Jan 27, 2026Updated 3 weeks ago
- Introducing dendrites to spiking neural networks. Designed for the Brian 2 simulator.☆42Jan 15, 2026Updated last month
- Learning with spiking neural networks using STDP synapses. Python, PyNN, NEST-simulator, numpy☆10Aug 25, 2020Updated 5 years ago
- EfficientFlow: Efficient Equivariant Flow Policy Learning for Embodied AI☆23Jan 17, 2026Updated last month
- Dynamical Systems with JAX☆12Jan 11, 2026Updated last month
- Flexible Inference for Predictive Coding Networks in JAX.☆65Jan 12, 2026Updated last month
- Virtual Bash interpreter with a virtual file system for multi-tenant environments.☆31Updated this week
- berg 🦀 Transform the contents of Epub documents.☆10Apr 27, 2023Updated 2 years ago
- Testing Difference Target Propagation (DTP) on MNIST.☆12Oct 12, 2020Updated 5 years ago
- The QGIS plugin provides access to the WMS services of the Spatial Data Infrastructure TerraBrasilis.☆12Oct 24, 2025Updated 3 months ago
- Prototype for sparse automatic differentiation in JAX + Julia☆20Jun 17, 2025Updated 7 months ago
- Jet tagging in the Lund plane with graph networks☆10Dec 16, 2021Updated 4 years ago
- JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library☆12Oct 12, 2025Updated 4 months ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆14May 28, 2025Updated 8 months ago
- Python implementation of Predictive Coding Networks and Predictive Coding Graphs, using PyTorch.☆16Jul 22, 2024Updated last year
- A nascent Jax-based package for virtual brain modeling.☆12Feb 3, 2026Updated last week
- LONGAGENT: Scaling Language Models to 128k Context through Multi-Agent Collaboration☆11Mar 11, 2024Updated last year
- Translate word embeddings across models☆10Aug 17, 2020Updated 5 years ago
- Decoupled Neural Interfaces Using Synthetic Gradients - under develeopment☆11Jun 27, 2025Updated 7 months ago
- 3rd placed submission to the NeurIPS MineRL competition 2019☆10Mar 24, 2023Updated 2 years ago
- ☆13Dec 5, 2022Updated 3 years ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆44Sep 7, 2023Updated 2 years ago