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

Commit

11 days ago

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

References

Author

danielsuo

Committer

a-googler

Parents