Skip to content

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

New pull request New
9 Open 161 Closed
9 Open 161 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

Update table_stacking_test and embedding_test to work for tpuv6.
#170 opened Jan 23, 2025 by copybara-service bot Loading…
Internal
#169 opened Jan 23, 2025 by copybara-service bot Loading…
Input preprocessing library to support vocab-dimension mini-batching.
#165 opened Jan 22, 2025 by copybara-service bot Loading…
Prototype integration of vocab-dimension mini-batching into trainer_lib.
#160 opened Jan 21, 2025 by copybara-service bot Loading…
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
#149 opened Jan 13, 2025 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.