Rewritten PyTorch framework designed to help you learn AI/ML
Project description
edutorch
Rewritten PyTorch framework designed to help you learn AI/ML!
PyTorch is one of the most amazing frameworks for building and training deep neural networks. One of its biggest strengths is providing an intuitive and extendable interface for building and training these models.
In this project, I provide my own version of the PyTorch framework, designed to help you understand the key concepts. The goal is to provide explicit implementations of popular layers, models, and optimizers. Above all else, this code is designed to be readable and clear. Many of these examples are modified from Stanford's CS 230 / 231N course materials available online.
EduTorch vs PyTorch
One notable difference between EduTorch and PyTorch is that EduTorch does NOT provide autograd. There are many educational benefits to deriving/implementing the backprop step yourself, and if you want automatic gradient calculations, you are better off using the real framework. Additionally, if you wanted to learn how the autograd system is implemented, you can check out Andrej Karpathy's micrograd project
There is no CUDA or GPU support either, for the same reasons.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.