| # 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/. | |