flax
[flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit.
#5125

Merged

[flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit. #5125

copybara-service merged 1 commit into main from test_838909002

copybara-service force pushed from 62ee4c2c to 15732157 16 days ago

[flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit.

881b8ed4

copybara-service force pushed from 15732157 to 881b8ed4 16 days ago

copybara-service merged 881b8ed4 into main 16 days ago

copybara-service deleted the test_838909002 branch 16 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone