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

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