rsokl / MyGrad

Drop-in autodiff for NumPy.
https://rsokl.github.io/MyGrad/
MIT License
194 stars 21 forks source link

Simplify back-prop scheme #407

Closed rsokl closed 2 years ago

rsokl commented 2 years ago

Traversing the graph based on tensors, not operations, makes things much nicer!