Spyx: Spiking Neural Networks in JAX
A flexible trainer interface for Jax and Haiku.
Haiku-based Hypernetworks in JAX
JAX-based Model Explanation and Interpretation Library
