jax
6aa35a2a - Adding an option to return the output tree in make_jaxpr

Commit

5 years ago

Adding an option to return the output tree in make_jaxpr

References

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

Author

botev

botev

Committer

botev

botev

Parents

Loading