torch>=2.4
torchvision>=0.19
numpy>=1.26
einops>=0.7
kornia>=0.8
timm>=0.9
tqdm>=4.66
pyyaml>=6.0
matplotlib>=3.8

[all]
revit-pde[datasets,dev,train,tune]

[datasets]
h5py>=3.10
the-well

[dev]
pytest>=8

[train]
tensorboard

[tune]
ray[tune]
