jaxtyping>=0.3.3
beartype>=0.22.2
torch>=2.2.0
lightning-utilities>=0.15.2
pydantic>=2.12.4
torchvision>=0.24.1
einops>=0.8.1
polars>=1.37.0
pyarrow>=22.0.0
scipy>=1.15.3
tqdm>=4.67.1

[notebook]
jupyter>=1.1.1

[onnx]
onnx>=1.19.1
onnxscript>=0.5.3

[tensorrt-cu12]
tensorrt-cu12==10.9.0.34

[transformers]
transformers>=4.57.1

[triton-client]
tritonclient[all]<=2.60.0
