Skip to content

A ML library in the middle between Pytorch and Micrograd

License

Notifications You must be signed in to change notification settings

mattianeroni/MedioGrad

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MedioGrad (WIP)

A ML library in the middle between Pytorch and Micrograd.

Not evoluted as TinyGrad since objective is to be numpy-based only (maybe in future with a bit of pycuda).

Stupid example

from tensor import Tensor 
import graph

a = Tensor([1,2,3])
b = Tensor([1,1,1])
c = a + b
c.backward()
graph.draw_dot(c, format="png").render(filename="graph")

alt text

About

A ML library in the middle between Pytorch and Micrograd

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages