rockpool.devices.dynapse.quantization.autoencoder.gradient

Dynap-SE autoencoder based quantization package custom gradient implementation

  • Non User Facing *

Functions

step_pwl_jvp(primals, tangents)

step_pwl_jvp custom jvp function defining the custom gradient rule of the step pwl function