torch~=2.2.0
torchvision~=0.17.0
einops~=0.7.0
pydantic~=2.0
wandb~=0.16.2
fire~=0.5.0
tqdm~=4.66.1
pytest~=8.1.2
ipykernel~=6.29.0
transformer-lens~=1.14.0
jaxtyping~=0.2.25
python-dotenv~=1.0.1
zstandard~=0.22.0
matplotlib~=3.5.3
seaborn~=0.13.2
tenacity~=8.2.3
statsmodels~=0.14.2

[dev]
ruff~=0.1.14
pyright~=1.1.360
pre-commit~=3.6.0
