jax
b8396c1d - Merge pull request #4364 from botev:api

Commit

5 years ago

Merge pull request #4364 from botev:api PiperOrigin-RevId: 333478420

References

#4364 - Adding an option to return the output tree in make_jaxpr

Author

a-googler

a-googler

Parents

Loading