numpy>=1.24.0
scipy>=1.10.0

[coreml]
coremltools>=7.0

[cuda]
cupy-cuda12x>=12.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
ruff>=0.1.0

[full]
scipy>=1.10.0
pyarrow>=12.0.0
h5py>=3.8.0
netCDF4>=1.6.0
matplotlib>=3.7.0
numba>=0.57.0
triton>=2.1.0
gymnasium>=0.29.0
onnx>=1.14.0
onnxruntime>=1.16.0
huggingface-hub>=0.20.0
safetensors>=0.4.0

[io]
pyarrow>=12.0.0
h5py>=3.8.0
netCDF4>=1.6.0

[numba]
numba>=0.57.0

[onnx]
onnx>=1.14.0
onnxruntime>=1.16.0

[pretrained]
huggingface-hub>=0.20.0
safetensors>=0.4.0

[rl]
gymnasium>=0.29.0

[tokenizers]
tokenizers>=0.15.0

[triton]
triton>=2.1.0

[viz]
matplotlib>=3.7.0
