← all repositories

jcjohnson/pytorch-examples

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

4.9k stars Python ML FrameworksLearning
pytorch-examples
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.