flax
Track explicit sharding in Variables
#5245
Open

Track explicit sharding in Variables #5245

samanklesaria
gemini-code-assist
cgarciae
cgarciae commented on 2026-02-12
samanklesaria samanklesaria force pushed from aca3ded4 to 9a11f90d 29 days ago
samanklesaria samanklesaria force pushed from 9a11f90d to 5f7e92af 29 days ago
samanklesaria samanklesaria force pushed from 5f7e92af to 2cb1e38f 29 days ago
samanklesaria samanklesaria requested a review from cgarciae cgarciae 29 days ago
samanklesaria samanklesaria force pushed from 2cb1e38f to 3af418b3 29 days ago
samanklesaria samanklesaria force pushed from 3af418b3 to f4430b78 29 days ago
samanklesaria samanklesaria force pushed from f4430b78 to 17ba19f6 29 days ago
cgarciae
cgarciae commented on 2026-03-04
cgarciae
cgarciae commented on 2026-03-04
samanklesaria Track explicit sharding in Variables
9919a36a
samanklesaria Add support for setting values with different sharding
6e422dce
samanklesaria Add error message when changing sharding
eac30b2f
samanklesaria samanklesaria force pushed from eecb1131 to eac30b2f 7 days ago
samanklesaria Remove unnecessary check for sharding
7f954fa3

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone