absl-py
torch==2.9.0
torchvision==0.24.0
torchaudio==2.9.0
torchao
--pre
ai-edge-litert-nightly==2.2.0.dev20260210
ai-edge-quantizer-nightly
jax[cpu]
torch-xla2[odml]>=0.0.1.dev20241201
scipy
numpy
tabulate
safetensors
kagglehub
transformers
multipledispatch
jaxtyping
fire