IPython
jaxtyping
matplotlib
numpy
pandas
seaborn
scipy
torch
torchaudio
torchvision
wandb
# Type Checks
ruff
mypy>=1.0
pytest
types-psutil
types-tqdm
types-PyYAML
