dm-haiku
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
