torch<2.6.0
torchvision
pydantic
wandb
fire
tqdm
pytest
ipykernel
jaxtyping
tiktoken
transformers
datasets
python-dotenv

[dev]
ruff
pyright
pre-commit
