torch
einops==0.8.1
h5py
pyyaml
matplotlib
diffusers
accelerate
torchjd
pydantic
pydantic_numpy
wandb
experiment-launcher-meco

[docs]
sphinx>=7.0
furo
sphinx-autodoc-typehints
sphinx-copybutton
myst-parser
