jcjohnson/pytorch-examples
Self-contained examples teaching PyTorch fundamentals including tensors, autograd, neural network modules, and gradient descent training.

Velocity · 7d
+1.4
★ / day
Trend
→steady
star history
This repository walks through PyTorch’s core features through a hands-on running example: training a fully-connected ReLU network with a single hidden layer using gradient descent to fit random data. It covers tensors, automatic differentiation, nn modules, optimizers, and custom autograd functions. It also includes comparable numpy and TensorFlow implementations for context.