Spyx SHD Paper benchmark needs reviewed

kmheckel / spyx

Spyx: Spiking Neural Networks in JAX

https://spyx.readthedocs.io/en/latest/

MIT License

103 stars 12 forks source link

Spyx SHD Paper benchmark needs reviewed #30

Open kmheckel opened 7 months ago

kmheckel commented 7 months ago

There might be a bug with respect to time_major = False or True that could present incorrect results since scanning over one axis is faster than scanning over the other (time steps vs channel dim...)