Skip to content

stefpi/stefgrad

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

stef grad

remake of micrograd and then expanding on micrograd

  • micrograd with scalar values
  • micrograd extended to tensors
  • add new loss functions
    • square loss error
    • cross entropy loss
  • add new activations functions
    • tanh
    • sigmoid
    • softmax
  • MNIST
    • train MNIST
    • train MNIST with batching
  • add GPU training support
  • train MNIST on GPU

About

a python autograd engine

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages