Skip to content

Pull requests: AI-Hypercomputer/RecML

New pull request New
18 Open 45 Closed
18 Open 45 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

[pallas] Do not use pl.{load,store} which will soon be removed from Pallas
#66 opened Aug 1, 2025 by copybara-service bot Loading…
Replace jax.sharding.use_mesh with jax.set_mesh. jax.set_mesh can act as a global setter or a context manager.
#65 opened Jul 31, 2025 by copybara-service bot Loading…
This CL introduces an end-to-end Flax implementation of the DLRM-HSTU model, along with individual unit tests for the various components.
#64 opened Jul 31, 2025 by copybara-service bot Loading…
@chandrasekhard2
[pallas] Do not use pl.{load,store} which will soon be removed from Pallas
#63 opened Jul 30, 2025 by copybara-service bot Loading…
Refactor a few APIs.
#62 opened Jul 30, 2025 by copybara-service bot Loading…
[pallas] Do not use pl.{load,store} which will soon be removed from Pallas
#61 opened Jul 24, 2025 by copybara-service bot Loading…
Bump protobuf from 4.21.12 to 4.25.8 dependencies Pull requests that update a dependency file python Pull requests that update python code
#56 opened Jun 26, 2025 by dependabot bot Loading…
Updating Documentation and config
#51 opened Jun 19, 2025 by depksingh Loading…
2
Bump urllib3 from 2.4.0 to 2.5.0 dependencies Pull requests that update a dependency file python Pull requests that update python code
#50 opened Jun 19, 2025 by dependabot bot Loading…
Save variables with keys in the checkpoint
#46 opened Jun 14, 2025 by copybara-service bot Loading…
Bump requests from 2.32.3 to 2.32.4 dependencies Pull requests that update a dependency file python Pull requests that update python code
#44 opened Jun 10, 2025 by dependabot bot Loading…
Bump torch from 2.6.0 to 2.7.1 dependencies Pull requests that update a dependency file python Pull requests that update python code
#42 opened Jun 7, 2025 by dependabot bot Loading…
Add back pyproject.toml and uv.lock and use uv in actions.
#39 opened May 21, 2025 by Hilly12 Loading…
Bump setuptools from 78.1.0 to 78.1.1 dependencies Pull requests that update a dependency file python Pull requests that update python code
#36 opened May 20, 2025 by dependabot bot Loading…
Internal Change
#31 opened Apr 23, 2025 by copybara-service bot Loading…
1
Internal change.
#30 opened Apr 23, 2025 by copybara-service bot Loading…
Add DLRM-V2 with sparsecore.
#20 opened Apr 13, 2025 by copybara-service bot Loading…
Add Keras trainer.
#15 opened Apr 9, 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.