jatinparab98 / chotu-grad Goto Github PK
View Code? Open in Web Editor NEWA very small version of pytorch's autograd class. Calculate gradients of scalars with forward and backward pass. Can be used to train small MLPs, mostly for educational purpose.