flax
Check for leaves of type variablelib.Variable when getting sharding specs.
#4769
Merged

Check for leaves of type variablelib.Variable when getting sharding specs. #4769

copybara-service merged 1 commit into main from test_766401557
copybara-service
google-cla
copybara-service copybara-service force pushed from b6bf5bb7 to 59acb034 1 year ago
copybara-service copybara-service force pushed from 59acb034 to 05370ae9 1 year ago
copybara-service copybara-service force pushed from 05370ae9 to d743ec77 1 year ago
Check for leaves of type variablelib.Variable when getting sharding s…
096b8a02
copybara-service copybara-service force pushed from d743ec77 to 096b8a02 1 year ago
copybara-service copybara-service merged 096b8a02 into main 1 year ago
copybara-service copybara-service deleted the test_766401557 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone