torch>=2.6.0
torchvision
torchaudio
einops
h5py
matplotlib
numpy
packaging
pydantic
requests
SciencePlots
scipy
tensorboard
tqdm

[cupy11]
cupy-cuda11x

[cupy12]
cupy-cuda12x

[lightning]
lightning
jsonargparse[signatures]

[nir]
nir
nirtorch

[triton]
triton>=3.3.1
