requirements.txt

# Python 3.11 or later.
#
# Lectures 1, 4, 5, 6 and 7 need only torch and numpy, and run on a laptop CPU.
# Lectures 2 and 3 need torchvision for MNIST, and transformers for the ESM-2 examples.
torch>=2.4
numpy>=1.26
scipy>=1.11
torchvision>=0.19   # Lectures 2 and 3: MNIST and image grids.
transformers>=4.40  # Lecture 3 only, and only with --dataset esm2.
pytest>=8.0         # The test suite under tests/.