Skip to content

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

New pull request New
18 Open 301 Closed
18 Open 301 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
#319 opened Jun 23, 2025 by copybara-service bot Loading…
[JAX SC] Clean up tf.sparse.tensor preprocessing lib.
#309 opened Jun 18, 2025 by copybara-service bot Loading…
Access EmbeddingVariables table instead of relying on indexing.
#302 opened Jun 17, 2025 by copybara-service bot Loading…
[JAX SC] Migrate to unified preprocessing library.
#298 opened Jun 13, 2025 by copybara-service bot Loading…
Added fallback logic for computing COO buffer size
#294 opened Jun 12, 2025 by copybara-service bot Loading…
Expose run‑time FDO limits through FdoParams and InputParams in the API.
#293 opened Jun 12, 2025 by copybara-service bot Loading…
Do not count the sentinel node for checking the buffer overflow as it is not actually filled in the COO buffer.
#263 opened May 27, 2025 by copybara-service bot Loading…
Introduce FdoStats struct for tracking FDO info.
#253 opened May 12, 2025 by copybara-service bot Loading…
Refactor to use a config object to manage arguments to sparse dense matmul.
#235 opened Apr 25, 2025 by copybara-service bot Loading…
Internal Change
#230 opened Apr 23, 2025 by copybara-service bot Loading…
Skip tests with incorrect topology
#227 opened Apr 16, 2025 by copybara-service bot Loading…
Googly changes
#205 opened Mar 5, 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! Adding no:label will show everything without a label.