Unpin JAX version after next jax_tpu_embedding release
Open
- Dominant language
- Python
- Stars
- 64
- Forks
- 18
- Avg merge
- 4h 46m
- Merged PRs (30d)
- 1
Description
Pinned in https://github.com/google/metrax/pull/106 to fix import errors in tests.
Contributor guide
Assessment
This issue has not been assessed yet.