A PyTorch API implemented from scratch over the course of 5 modules. Scalar operations, tensor operations, convolutions have been implemented both for forward propagation in addition to the backpropagation. Tensor operations have been optimized for both the CPU and for CUDA
-
Notifications
You must be signed in to change notification settings - Fork 0
NamanMakkar/minitorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
minitorch - A PyTorch API implemented from scratch for the course CS5781 Machine Learning Engineering
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published