Spyx SHD Paper benchmark needs reviewed

kmheckel / spyx

Spyx: Spiking Neural Networks in JAX

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

MIT License

98 stars 11 forks source link

Spyx SHD Paper benchmark needs reviewed #30

Open kmheckel opened 4 months ago

kmheckel commented 4 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...)