SpikingJelly

中文 | English

Contents

• Why SpikingJelly
• Installation
• Quick Start
• Core Capabilities
  • Backend Performance
  • Large-Scale SNN Systems
  • Datasets
  • Interchange and Deployment
• Project Status and Version Notes
• License
• Acknowledgement
• Contributing
• Citation

Why SpikingJelly

SpikingJelly is a PyTorch-native framework for spiking neural networks (SNNs), with support for large-scale SNN training and inference.

• Beginner-friendly API
• ANN2SNN conversion
• Event-based datasets
• Acceleration backends: torch, cupy, triton
• Memory-efficient training, distributed execution, precision control
• Hardware deployment and framework exchange

Installation

SpikingJelly is built on PyTorch. Install PyTorch, torchvision, and torchaudio first.

• Python >=3.11
• PyTorch >=2.6.0 (tested with 2.7.1)

Install the latest stable PyPI release:

pip install spikingjelly

Install V2 pre-releases from PyPI when they are published:

pip install --pre spikingjelly

Install the latest development version from source:

git clone https://github.com/fangwei123456/spikingjelly.git
cd spikingjelly
pip install .

Optional dependencies:

Quick Start

Define an SNN in the same way that you would define any PyTorch model:

from torch import nn
from spikingjelly.activation_based import layer, neuron, surrogate

net = nn.Sequential(
    layer.Flatten(),
    layer.Linear(28 * 28, 10, bias=False),
    neuron.LIFNode(tau=2.0, surrogate_function=surrogate.ATan())
)

Next steps:

• Learn about spiking neurons
• Train on event-based datasets
• Convert ANN to SNN

Core Capabilities

Backend Performance

Spiking neuron models run on torch, cupy, or triton backends. The backend is set at neuron creation and can be changed later. All backends are compatible with torch.compile.

Below: execution time comparison for multi-step LIF neurons on torch vs cupy. Triton is covered in the backend tutorials.

Large-Scale SNN Systems

For large-scale SNN systems, SpikingJelly provides:

• Memory-efficient training with spike compression (memopt)
• Experimental distributed execution for multi-GPU workloads
• Precision policy tools for large-scale training and inference
• Spiking transformer components

Datasets

SpikingJelly includes the following event-based and neuromorphic datasets:

• ASL-DVS
• Bullying10K
• CIFAR10-DVS
• DVS-Lip
• DVS128 Gesture
• ES-ImageNet
• HARDVS
• N-Caltech101
• N-MNIST
• Nav Gesture
• SHD
• SSC
• Speech Commands

Each dataset supports raw event access and frame representations. See the neuromorphic datasets tutorial for the full workflow.

Interchange and Deployment

Export SpikingJelly models to neuromorphic hardware or other frameworks:

• NIR exchange
• Lava exchange
• Lynxi deployment

Project Status and Version Notes

Development / release policy:

Starting from SpikingJelly V2, release versions follow PEP 440 compatible, SemVer-style MAJOR.MINOR.PATCH semantics. MAJOR marks compatibility generations, MINOR adds backward-compatible functionality, and PATCH fixes bugs. Python package pre-release spelling is used for V2 development releases, for example 2.0.0.dev0, 2.0.0a1, 2.0.0b1, and 2.0.0rc1.

See CHANGELOG.md for the V2 release changelog.

License

SpikingJelly is distributed under the Open-Intelligence Open Source License Version 1.0. See the authoritative Chinese license, translations, and the usage and distribution guide.

Acknowledgement

Maintainers

Current maintainers (since July 2024):

• Yifan Huang
• Peng Xue

Previous core maintainers (before July 2024):

• Wei Fang
• Yanqi Chen
• Jianhao Ding
• Ding Chen
• Liwei Huang

Contributors

The full contributor list is on the contributors page.

Institutes

The main institutions behind SpikingJelly are Multimedia Learning Group, Institute of Digital Media (NELVT), Peking University and Peng Cheng Laboratory.

Community and related links

• Documentation
• Contributing Guide
• Issues and development discussion
• OpenI mirror
• Community Jupyter tutorials in Chinese

Contributing

We welcome issues, pull requests, documentation improvements, and translations.

• Read the Contributing Guide.
• Check issues for ongoing work.
• API docs are not fully bilingual yet; translation contributions are especially welcome.

Citation

Publications using SpikingJelly are listed on the documentation page. The source of truth in this repository is publications.json.

If you use SpikingJelly in your work, please cite:

@article{
doi:10.1126/sciadv.adi1480,
author = {Wei Fang and Yanqi Chen and Jianhao Ding and Zhaofei Yu and Timothee Masquelier and Ding Chen and Liwei Huang and Huihui Zhou and Guoqi Li and Yonghong Tian},
title = {SpikingJelly: An open-source machine learning infrastructure platform for spike-based intelligence},
journal = {Science Advances},
volume = {9},
number = {40},
pages = {eadi1480},
year = {2023},
doi = {10.1126/sciadv.adi1480},
url = {https://www.science.org/doi/abs/10.1126/sciadv.adi1480},
eprint = {https://www.science.org/doi/pdf/10.1126/sciadv.adi1480}
}