snntorch

收录理由

对于构建脉冲神经网络的研究者和工程师来说，snntorch 让你不必离开熟悉的 PyTorch 环境。它把每个脉冲神经元建模成标准 torch.nn 风格模型中的循环激活单元，因此你依然可以使用熟悉的层、自动求导和 CUDA 加速，同时探索基于脉冲的计算方式。这个库内置了多种神经元模型、用于训练的代理梯度函数、脉冲生成与转换工具，以及可视化和数据集辅助功能，省去了许多通常拖慢 SNN 研究的样板代码。它还支持通过 NIR 交换格式导入和导出模型，方便在不同脉冲框架之间迁移网络。无论是做神经形态计算还是能效计算，教程和 Colab 笔记本都是不错的起点，需要时也能在 CPU 上跑小型训练任务。