spyx.axn
July 3, 2026 ยท View on GitHub
Surrogate-gradient factories that return JIT-compiled jax.custom_gradient functions. Pass the returned callable as the activation argument to any spiking neuron.
::: spyx.axn
July 3, 2026 ยท View on GitHub
Surrogate-gradient factories that return JIT-compiled jax.custom_gradient functions. Pass the returned callable as the activation argument to any spiking neuron.
::: spyx.axn