Skip to content

Pull requests: AI-Hypercomputer/RecML

New pull request New
38 Open 49 Closed
38 Open 49 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

Add compatibility for new fiddle version.
#90 opened Sep 22, 2025 by copybara-service bot Loading…
Update TF Data service flags.
#89 opened Sep 19, 2025 by copybara-service bot Loading…
1
Bump keras from 3.9.2 to 3.11.3 dependencies Pull requests that update a dependency file python Pull requests that update python code
#88 opened Sep 19, 2025 by dependabot bot Loading…
Add compatibility for new fiddle version.
#87 opened Sep 19, 2025 by copybara-service bot Loading…
Update TF Data service flags.
#85 opened Sep 17, 2025 by copybara-service bot Loading…
1
Update TF Data service flags.
#84 opened Sep 16, 2025 by copybara-service bot Loading…
@chandrasekhard2
Update TF Data service flags.
#83 opened Sep 13, 2025 by copybara-service bot Loading…
1
Integrate sparsecore into DLRM HSTU implementation. Additionally implemented KV caching for STU layers
#81 opened Sep 12, 2025 by copybara-service bot Loading…
@chandrasekhard2
Integrate sparsecore into DLRM HSTU implementation. Additionally implemented KV caching for STU layers
#80 opened Sep 12, 2025 by copybara-service bot Loading…
@chandrasekhard2
Integrate sparsecore into DLRM HSTU implementation. Additionally implemented KV caching for STU layers
#79 opened Sep 12, 2025 by copybara-service bot Loading…
@chandrasekhard2
Remove compile config from new checkpoint format.
#78 opened Sep 12, 2025 by copybara-service bot Loading…
Integrate sparsecore into DLRM HSTU implementation. Additionally implemented KV caching for STU layers
#77 opened Sep 11, 2025 by copybara-service bot Loading…
@chandrasekhard2
Integrate sparsecore into DLRM HSTU implementation. Additionally implemented KV caching for STU layers
#76 opened Sep 11, 2025 by copybara-service bot Loading…
@chandrasekhard2
Integrate sparsecore into DLRM HSTU implementation. Additionally implemented KV caching for STU layers
#75 opened Sep 8, 2025 by copybara-service bot Loading…
@chandrasekhard2
Adds initial Keras Orbax checkpointer V2 implementation.
#74 opened Sep 2, 2025 by copybara-service bot Loading…
Integrate sparsecore into DLRM HSTU implementation. Additionally implemented KV caching for STU layers
#73 opened Aug 25, 2025 by copybara-service bot Loading…
@chandrasekhard2
[Efficient LMs] Add compressor and decompressor for the research work of go/context_compression. This commit adds functions for compressing and decompressing input and output tensors.
#72 opened Aug 15, 2025 by copybara-service bot Loading…
1
Integrate sparsecore into DLRM HSTU implementation. Additionally implemented KV caching for STU layers
#71 opened Aug 14, 2025 by copybara-service bot Loading…
Reverts changelist 793734230
#70 opened Aug 13, 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.
#68 opened Aug 7, 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.
#67 opened Aug 6, 2025 by copybara-service bot Loading…
[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
Refactor a few APIs.
#62 opened Jul 30, 2025 by copybara-service bot Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.