Skip to content

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

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

Change type annotation for barnacore_embedding_sharding
#136 opened Oct 10, 2024 by copybara-service bot Loading…
1
Sort devices explicitly by process index, then id (as opposed to IDs alone). IDs may be randomly generated, and are not guaranteed to be ordered based on their process index.
#123 opened Jul 10, 2024 by copybara-service bot Loading…
1
Remove the OCDBT check for now.
#105 opened May 13, 2024 by copybara-service bot Loading…
1
Internal.
#100 opened May 2, 2024 by copybara-service bot Loading…
Refactors input processing with packer to avoid extra structure traversals.
#98 opened Apr 25, 2024 by copybara-service bot Loading…
1
Code update
#93 opened Apr 24, 2024 by copybara-service bot Loading…
Replace references to deprecated device_buffer attributes
#67 opened Dec 6, 2023 by copybara-service bot Loading…
testing
#37 opened Apr 20, 2023 by khshah6 • Draft
add reqs for pypi package
#36 opened Apr 20, 2023 by copybara-service bot Loading…
ProTip! Filter pull requests by the default branch with base:main.