jax
af50c212 - Remove deprecated API and migrate to new API.

Commit

1 year ago

Remove deprecated API and migrate to new API. Context: https://github.com/jax-ml/jax/pull/21716 P.S. minor formatting fixes. PiperOrigin-RevId: 684896546

References

#24213 - Remove deprecated API and migrate to new API.

Author

ghpvnist

Committer

Google-ML-Automation

Parents

a684021c

jax af50c212 - Remove deprecated API and migrate to new API.

jax
af50c212 - Remove deprecated API and migrate to new API.