torch
torchvision
einops
pydantic
wandb
fire
tqdm
pytest
ipykernel
transformer-lens
jaxtyping
python-dotenv
zstandard
matplotlib
seaborn
umap-learn
tenacity
statsmodels
automated-interpretability

[dev]
ruff
pyright
pre-commit
