search

kmheckel / spyx

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

Add Type Checking via Chex #10

Open kmheckel opened 11 months ago

kmheckel commented 11 months ago

Use Chex to add type checking to the library in spots not already handled by Haiku.