
[jax]
deprecated>=1.2.13
jax>=0.8.3
keras>=3.13.2
keras-hub>=0.26.0
numpy
opencv-python-headless
Pillow
prettytable
psutil
py-cpuinfo
pydantic
schema
scikit-learn
sentencepiece
tokenizers

[jax:sys_platform != "win32" or python_version > "3.8"]
pycocotools

[jax:sys_platform == "win32" and python_version <= "3.8"]
pycocotools-windows

[pt]
numpy
packaging
prettytable
psutil
py-cpuinfo
pydantic
torchao

[tf]
prettytable
psutil
py-cpuinfo
pydantic
pyyaml
tensorflow
