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
pyarrow-stubs>=20.0.0.20251215
scipy-stubs>=1.15.3.0
types-tqdm>=4.67.3.20260303

[notebook]
jupyter>=1.1.1

[onnx]
onnx>=1.19.1
onnxscript>=0.5.3

[onnx:sys_platform != "linux"]
onnxruntime>=1.24.3

[onnx:sys_platform == "linux"]
onnxruntime-gpu>=1.24.3

[transformers]
transformers>=5.3.0
