-
Notifications
You must be signed in to change notification settings - Fork 3
Pull requests: jax-ml/jax-tpu-embedding
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[JAX SC] [23/n] [Minibatching] Add gRPC-based AllReduce implementation for SparseCore.
#440
opened Aug 22, 2025 by
copybara-service
bot
Loading…
Refactor input preprocessing data structures.
#436
opened Aug 21, 2025 by
copybara-service
bot
Loading…
Access
EmbeddingVariables table instead of relying on indexing.
#302
opened Jun 17, 2025 by
copybara-service
bot
Loading…
Added fallback logic for computing COO buffer size
#294
opened Jun 12, 2025 by
copybara-service
bot
Loading…
Fix incorrect number of return values in docstring of embedding.preprocess_sparsecore_inputs.
#198
opened Feb 27, 2025 by
copybara-service
bot
Loading…
[CI] Create tpu_presubmit.yml - testing
CI Connection Halt - On Retry
Flag every job that has its workflow file setup for halting to halt if the workflow is retried
#180
opened Feb 7, 2025 by
MichaelHudgins
•
Draft
Split several shared functions into utility modules.
#159
opened Jan 21, 2025 by
copybara-service
bot
Loading…
Input preprocessing library to support vocab-dimension mini-batching.
#155
opened Jan 18, 2025 by
copybara-service
bot
Loading…
Input preprocessing library to support vocab-dimension mini-batching.
#154
opened Jan 18, 2025 by
copybara-service
bot
Loading…
Add presubmit script to jax-tpu-embedding/sparsecore
CI Connection Halt - On Retry
Flag every job that has its workflow file setup for halting to halt if the workflow is retried
#149
opened Jan 13, 2025 by
copybara-service
bot
Loading…
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.