Gograd is a small automatic reverse-mode differentiation framework written in Go.
a := NewVar(4)
b := NewVar(3)
c := a.Mul(a.Add(b))
c.Backward()
fmt.Println(c.value)
// prints 28
fmt.Println(a.grad)
// prints 11
fmt.Println(b.grad)
// prints 4
This repository has been archived by the owner on Oct 20, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
Gograd is a small automatic differentiation framework written in Go.
License
lucaionescu/gograd
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Gograd is a small automatic differentiation framework written in Go.