Make jax_tpu_embedding utils reference non global to avoid dependency.
#55
test_775914875 was force-pushed and no longer has any new commits.
Pushing new commits will allow the pull request to be re-opened.