Skip to content

Pull requests: jax-ml/jax-tpu-embedding

New pull request New
8 Open 177 Closed
8 Open 177 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Fix embedding_test and table_stacking_test for tpuv6.
#184 opened Feb 10, 2025 by copybara-service bot Loading…
Internal changes
#182 opened Feb 10, 2025 by copybara-service bot Loading…
[CI] Create tpu_presubmit.yml - testing
#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…
Internal change only
#150 opened Jan 15, 2025 by copybara-service bot Loading…
1
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.