kmheckel / spyx

Spyx: Spiking Neural Networks in JAX
https://spyx.readthedocs.io/en/latest/
MIT License
98 stars 11 forks source link

Implement EXODUS algorithm #14

Closed kmheckel closed 7 months ago

kmheckel commented 10 months ago

Could implement EXODUS to drastically cut gradient calculation time for certain classes of models and improve training speed even further.

https://www.frontiersin.org/articles/10.3389/fnins.2023.1110444/full