tpu-info
yapf==0.43.0
pytest
pytest-mock
absl-py
numpy
google-cloud-storage
--pre
--index-url https://us-python.pkg.dev/ml-oss-artifacts-published/jax/simple/
--find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html
jax==0.7.1
jaxlib==0.7.1
libtpu==0.0.23
jaxtyping
flax==0.11.1
torchax==0.0.5
qwix
pathwaysutils
