kmheckel / spyx

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

Add Phasor layers/network capability #13

Open kmheckel opened 9 months ago

kmheckel commented 9 months ago

It would be awesome to implement the ability to train spiking phasor networks in Spyx. JAX has the ability to support complex-valued autodifferentiation, so this should be possible. Doing this would enable extremely fast training by eliminating recurrence when learning before converting to a recurrent architecture for inference.

https://arxiv.org/abs/2204.00507

kmheckel commented 5 months ago

This is tied to the implementation of latency based training tools/objectives. #21