flax
45c29559 - Support jax.experimental.array checkpointing.

Commit

3 years ago

Support jax.experimental.array checkpointing. PiperOrigin-RevId: 475669596

Author

IvyZX

IvyZX

Committer

a-googler

a-googler

Parents

Loading