Skip to content

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

New pull request New
10 Open 189 Closed
10 Open 189 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 incorrect number of return values in docstring of embedding.preprocess_sparsecore_inputs.
#198 opened Feb 27, 2025 by copybara-service bot Loading…
Update bazelrc to prepare for CI and release builds.
#197 opened Feb 27, 2025 by copybara-service bot Loading…
Minimize dependencies.
#196 opened Feb 26, 2025 by copybara-service bot Loading…
stack and shard feature table API
#194 opened Feb 21, 2025 by copybara-service bot Loading…
Fix autograd_test for tpuv6.
#188 opened Feb 12, 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! Adding no:label will show everything without a label.