issues
search
brainpy
/
BrainPy
Brain Dynamics Programming in Python
https://brainpy.readthedocs.io/
GNU General Public License v3.0
508
stars
92
forks
source link
[math] change the internal implementation of surrogate function
#562
Closed
chaoming0625
closed
8 months ago
chaoming0625
commented
8 months ago
Description
add
heaviside_p
primitive
all surrogate functions (with one argument) support both JVP (forward) and VJP (backward) differentiations
enabling the forward gradients of spiking neural networks
How Has This Been Tested
Yes.
Description
heaviside_p
primitiveHow Has This Been Tested
Yes.