-
Notifications
You must be signed in to change notification settings - Fork 3
Pull requests: jax-ml/jax-tpu-embedding
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Introduce FdoStats struct for tracking FDO info.
#253
opened May 12, 2025 by
copybara-service
bot
Loading…
Record and Use COO Buffer Usage Fraction Statistics. Also changes
static_buffer_size_multiplier to a float for finer grain control.
#243
opened May 5, 2025 by
copybara-service
bot
Loading…
Add support for measuring preprocessed input buffer usage.
#242
opened May 5, 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…
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!
Type g p on any issue or pull request to go back to the pull request listing page.