jax
0e5e5c75 - [hijax] rename NewstyleHiPrimitive, expose in jax.experimental

Commit

210 days ago

[hijax] rename NewstyleHiPrimitive, expose in jax.experimental Co-authored-by: Dougal Maclaurin dougalm@google.com PiperOrigin-RevId: 826269432

References

Author

mattjj

Committer

Google-ML-Automation

Parents