Skip to content

A toy tensor library I made using two brain cells

License

Notifications You must be signed in to change notification settings

samsafadi/twocells

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Twocells

I wrote this toy tensor library for practice/fun after work. I use numpy as the base for my Tensor class and implemented autograd. As of right now I defined enough classes in order to train a simple MNIST fully-connected classifier.

It has an SGD and Adam implementation as the optimizers and two loss functions.

You'll never need PyTorch again!

TODO:

  • Add CNNs
  • Add BatchNorm
  • Add Dropout
  • Probably a million other features

About

A toy tensor library I made using two brain cells

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages