Input preprocessing library to support vocab-dimension mini-batching. #155

copybara-service[bot] wants to merge 1 commit intomainjax-ml/jax-tpu-embedding:mainfrom

test_716024250jax-ml/jax-tpu-embedding:test_716024250Copy head branch name to clipboard