Skip to content

Latest commit

 

History

History
22 lines (16 loc) · 403 Bytes

README.md

File metadata and controls

22 lines (16 loc) · 403 Bytes

axgrad

Backpropagation engine for scalar values.

This code has been created just to train myself in how backpropagation works (get the proper intuition).

Example usage

from axgrad.axgrad import Value

for i in range(1, 10):
    x = Value(i)
    y = x**3 + 2*x**2 + x - 7
    y.backward(0.001)
    print(x, x.grad)

Run tests

python -m unittest tests/test_axgrad.py