flax
add nnx.map and nnx.abstract_with_sharding
#5366
Merged
Overview
Commits
1
Changes
View On GitHub

add nnx.map and nnx.abstract_with_sharding #5366

copybara-service merged 1 commit into main from test_885853313
copybara-service
copybara-service copybara-service force pushed from 32ce68ca to 8cd500cd 142 days ago
copybara-service copybara-service force pushed from 8cd500cd to 75d269ba 142 days ago
copybara-service copybara-service force pushed from 75d269ba to c0289544 141 days ago
copybara-service copybara-service force pushed from c0289544 to 516ab4f0 141 days ago
add nnx.map and nnx.abstract_with_sharding
1572c84e
copybara-service copybara-service force pushed from 516ab4f0 to 1572c84e 141 days ago
copybara-service copybara-service merged 1572c84e into main 141 days ago
copybara-service copybara-service deleted the test_885853313 branch 141 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum