mlcroissant>=1.0.0
numpy>=1.19
zstandard>=0.19.0

[all]
mlcroissant>=1.0.0
numpy>=1.19
zstandard>=0.19.0
pyarrow>=14
jax>=0.4
grain>=0.2

[all:sys_platform != "darwin" or platform_machine != "x86_64"]
torch>=2.3

[dev]
setuptools>=45
cython
numpy
pytest
pytest-cov
tomli

[jax]
jax>=0.4
grain>=0.2

[parquet]
pyarrow>=14

[pytorch]

[pytorch:sys_platform != "darwin" or platform_machine != "x86_64"]
torch>=2.3
