Skip to content

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

New pull request New
6 Open 568 Closed
6 Open 568 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Internal.
#574 opened Nov 17, 2025 by copybara-service bot Loading…
Optimize SparseCore input preprocessing by eliminating buffer copies.
#567 opened Nov 12, 2025 by copybara-service bot Loading…
[JAX SC] Refactor input preprocessing to optimize non-minibatching path. Remove bucketetization overhead when minibatching is disabled or in pass 1.
#541 opened Oct 31, 2025 by copybara-service bot Loading…
[JAX SC] required_buffer_size_per_sc is being undercounted when minibatching is enabled. The initial calculation does not account for the alignment padding added at the end of each minibatch.
#528 opened Oct 28, 2025 by copybara-service bot Loading…
Support 1D optimizer slot variables
#438 opened Aug 22, 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! Updated in the last three days: updated:>2025-11-15.