Intelligent-Computing-Lab-Yale / STAttenLinks
PyTorch Implementation of Spiking Transformer with Spatial-Temporal Attention (CVPR 2025)
☆28Updated 2 weeks ago
Alternatives and similar repositories for STAtten
Users that are interested in STAtten are comparing it to the libraries listed below
Sorting:
- [ECCV-24] Spiking Wavelet Transformer☆27Updated 5 months ago
- Codes of the paper: SpikingResformer: Bridging ResNet and Vision Transformer in Spiking Neural Networks (CVPR2024)☆59Updated 6 months ago
- Offical implementation of "Spike-driven Transformer V2: Meta Spiking Neural Network Architecture Inspiring the Design of Next-generation …☆183Updated last year
- Offical implementation of "Quantized Spike-driven Transformer" (ICLR2025)☆24Updated 2 months ago
- Offical implementation of "Spike-driven Transformer" (NeurIPS2023)☆260Updated last year
- Offical code of "QKFormer: Hierarchical Spiking Transformer using Q-K Attention" (NeurIPS 2024,Spotlight 3%)☆116Updated 5 months ago
- Membrane Potential Batch Normalization for Spiking Neural Networks☆19Updated last year
- Create a new backward path for more accurate SNN gradients.☆16Updated 10 months ago
- Spikingformer: Spike-driven Residual Learning for Transformer-based Spiking Neural Network☆104Updated 7 months ago
- ☆50Updated last year
- [ICCV2023] Towards Memory- and Time-Efficient Backpropagation for Training Spiking Neural Networks☆42Updated last year
- [TNNLS 2024] Implementation of "TCJA-SNN: Temporal-Channel Joint Attention for Spiking Neural Networks"☆56Updated last year
- SyOPs counter for spiking neural networks☆63Updated 2 years ago
- [ICLR 2024] Official implementation of Spiking Graph Contrastive Learning (0️⃣1️⃣ SpikeGCL)☆29Updated last year
- [ICASSP2022] RATE CODING OR DIRECT CODING: WHICH ONE IS BETTER FOR ACCURATE, ROBUST, and ENERGY-EFFICIENT SPIKING NEURAL NETWORKS☆19Updated last year
- Offical implementation of "Scaling Spike-driven Transformer with Efficient Spike Firing Approximation Training" (IEEE T-PAMI2025)☆70Updated last month
- ☆66Updated last week
- Joint A-SNN: Joint Training of Artificial and Spiking Neural Networks via Self-Distillation and Weight Factorization☆10Updated last year
- [Neurips 2023] Exploring Loss Functions for Time-based Training Strategy in Spiking Neural Networks☆15Updated last year
- This is the implentation of our paper "SpikeLLM: Scaling up Spiking Neural Network to Large Language Models via Saliency-based Spiking" i…☆22Updated 3 months ago
- AAAI-2024 "TC-LIF: A Two-Compartment Spiking Neuron Model for Long-Term Sequential Modelling"☆19Updated 10 months ago
- Offical Implementation of "CLIF: Complementary Leaky Integrate-and-Fire Neuron for Spiking Neural Networks" (ICML 2024 spotlight)☆21Updated last month
- GLIF: A Unified Gated Leaky Integrate-and-Fire Neuron for Spiking Neural Networks, NeurIPS 2022 Poster☆51Updated 2 years ago
- This project contains code for the paper titled "SpikingBERT: Distilling BERT to Train Spiking Language Models Using Implicit Differentia…☆23Updated last year
- Awesome Spiking Neural Networks☆134Updated 3 months ago
- Source code for AAAI 2025 paper: FSTA-SNN:Frequency-based Spatial-Temporal Attention Module for Spiking Neural Networks☆26Updated 4 months ago
- This is the implentation of our paper "SpikeLM: Towards General Spike-Driven Language Modeling via Elastic Bi-Spiking Mechanisms" in ICML…☆39Updated 11 months ago
- Deep Residual Learning in Spiking Neural Networks☆168Updated 2 years ago
- ☆62Updated last year
- [WACV 2024] Spiking Denoising Diffusion Probabilistic Models☆50Updated last month