tpu-info==0.7.1
yapf==0.43.0
pytest
pytest-mock
absl-py
numpy
google-cloud-storage
jax==0.11.0
jaxlib==0.11.0
libtpu==0.0.44
jaxtyping
fastapi[standard]<0.137.0
flax==0.12.8
torchax==0.0.13
qwix==0.1.2
torchvision==0.25.0
pathwaysutils
parameterized
numba==0.62.1
runai-model-streamer[gcs,s3]==0.15.9
gcsfs==2026.1.0
hypothesis==6.151.9
sortedcontainers==2.4.0
transformers==5.14.1
