torch>=2.6.0
torchvision
torchaudio
h5py
loguru<1,>=0.7
matplotlib
numpy
packaging
requests
SciencePlots
scipy
tensorboard
tqdm

[cupy11]
cupy-cuda11x

[cupy12]
cupy-cuda12x

[fp8-te]
transformer-engine[pytorch]<3,>=2.16

[fp8-torchao]
torchao<0.18,>=0.13

[lightning]
lightning
jsonargparse[signatures]

[nir]
nir
nirtorch

[qwen]
transformers==5.13.0

[triton]
triton>=3.3.1
