# Machine learning
numpy
torch

# Datasets and models
torchvision==0.21.0

# Experiments
lightning==2.5.1
torchmetrics==1.6.1
laplace-torch==0.2.2.2
scikit-learn==1.6.1

# Command line interface
fire

# Logging
wandb
tqdm

# Data wrangling
pandas

# Plotting
matplotlib
tueplots
seaborn>=0.12.2

# Hessians
hessian_eigenthings