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]
transformer-engine[pytorch]<3,>=2.16

[lightning]
lightning
jsonargparse[signatures]

[megatron]

[megatron:python_version >= "3.12"]
megatron-core==0.18.2

[nir]
nir<2,>=1.0.7
nirtorch<2.7,>=2.6

[qwen]
transformers==5.13.0

[sglang]

[sglang-export]
safetensors

[sglang:sys_platform == "linux"]
flash-attn-4==4.0.0b19
sglang==0.5.17

[triton]
triton>=3.3.1
