-
Notifications
You must be signed in to change notification settings - Fork 5
Pull requests: jax-ml/jax-tpu-embedding
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[JAX SC] Support quantization_config in SparseCore table stacking
#868
opened Aug 5, 2026 by
copybara-service
Bot
Loading…
Provide actual support for embedding dim=1 across TPU SparseCore optimizer grad primitives.
#856
opened Jul 29, 2026 by
copybara-service
Bot
Loading…
[JAX SC] Expand TPU CI test suite with hardware-compatible tests
#848
opened Jul 25, 2026 by
copybara-service
Bot
Loading…
[JAX SC] Add path exclusion list to TPU runtime test GitHub workflow
#847
opened Jul 25, 2026 by
copybara-service
Bot
Loading…
small bug fix and adding test on 1d embedding_table input
#845
opened Jul 24, 2026 by
copybara-service
Bot
Loading…
Support dimension=1 embedding tables on SparseCore.
#805
opened Jul 6, 2026 by
copybara-service
Bot
Loading…
Flax Linen: Support learning rate schedules
#742
opened Apr 2, 2026 by
copybara-service
Bot
Loading…
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.