Brain-inspired computing lab

SpikingBrain-7B. Spiking Brain-inspired Large Models, integrating hybrid efficient attention, MoE modules and spike encoding into its architecture

Spike-Driven-Transformer. Offical implementation of "Spike-driven Transformer" (NeurIPS2023)

SpikeYOLO. Offical implementation of "Integer-Valued Training and Spike-Driven Inference Spiking Neural Network for High-performance and Energy-efficient Object Detection" (ECCV2024 Best Paper Candidate)

Spike-Driven-Transformer-V2. Offical implementation of "Spike-driven Transformer V2: Meta Spiking Neural Network Architecture Inspiring the Design of Next-generation Neuromorphic Chips" (ICLR2024)

EMS-YOLO. Offical implementation of "Deep Directly-Trained Spiking Neural Networks for Object Detection" (ICCV2023)

Spike-Driven-Transformer-V3. Offical implementation of "Scaling Spike-driven Transformer with Efficient Spike Firing Approximation Training" (IEEE T-PAMI2025)

Attention-SNN. Offical implementation of "Attention Spiking Neural Networks" (IEEE T-PAMI2023)

Spike2Former. Offical implementation of "Spike2Former: Efficient Spiking Transformer for High-performance Image Segmentation" (AAAI2025 Oral)

MetaLA. Offical implementation of "MetaLA: Unified Optimal Linear Approximation to Softmax Attention Map" (NeurIPS2024 Oral)

ASA-SNN. Offical implementation of "Inherent Redundancy in Spiking Neural Networks" (ICCV2023)

SpikingBrain2.0. Offical implementation of "SpikingBrain2.0: Brain-Inspired Foundation Models for Efficient Long-Context and Cross-Platform Inference"

TA-SNN. Offical implementation of "Temporal-wise Attention Spiking Neural Networks for Event Streams Classification" (ICCV2021)

T-RevSNN. Offical implementation of High-Performance Temporal Reversible Spiking Neural Networks with $O(L)$ Training Memory and $O(1)$ Inference Cost, ICML 2024 spotlight.

MS-ResNet. Offical implementation of "Advancing Spiking Neural Networks towards Deep Residual Learning" (IEEE TNNLS 2024)

Spike-based-Dynamic-Computing. Offical implementation of "Spike-based dynamic computing with asynchronous sensing-computing neuromorphic chip" (Nature Communications 2024)

Int2Spike. Int2Spike matmul kernel

GAC. Offical implementation of "Gated Attention Coding for Training High-performance and Efficient Spiking Neural Networks" (AAAI2024)

RM-SNN. Offical implementation of "Sparser spiking activity can be better: Feature Refine-and-Mask spiking neural network for event-based visual recognition" (Neural Networks 2023)

MFS-Seg. Python