numpy
torch
tqdm
matplotlib
rich

[all]
wandb

[dev]
pytest
ruff
mypy

[wandb]
wandb
