jaxsnn.discrete.modules

Modules

`jaxsnn.discrete.modules.leaky_integrate`
`jaxsnn.discrete.modules.leaky_integrate_and_fire`

Functions

jaxsnn.discrete.modules.LI(out_dim, scale_in=0.2): Layer constructor function for a li (leaky-integrated) layer.

jaxsnn.discrete.modules.LIF(out_dim, method=<jax._src.custom_derivatives.custom_vjp object>, scale_in=0.7, scale_rec=0.2): Layer constructor function for a lif (leaky-integrated-fire) layer.