jax
e220bc25 - [trace_to_jaxpr] Refactor mosaic/core.py for trace_to_jaxpr.

Commit

61 days ago

[trace_to_jaxpr] Refactor mosaic/core.py for trace_to_jaxpr. PiperOrigin-RevId: 934399420

References

Author

gnecula

Committer

Google-ML-Automation

Parents