jaxsnn.discrete.threshold

Functions

jaxsnn.discrete.threshold.heaviside(x)

jaxsnn.discrete.threshold.superspike_bwd(res, g)

jaxsnn.discrete.threshold.superspike_fwd(x, alpha)