kmheckel / spyx

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

Make AEStream example #19

Open kmheckel opened 8 months ago

kmheckel commented 8 months ago

Test and make an example using AEStream to load data into Spyx.

An interesting question is whether AEStream can write to JAX tensors being used by Spyx in a compiled loop?

https://github.com/aestream/aestream/tree/v0.6.2