jax
12036017 - Merge pull request #726 from google/dont-pack-all-outputs

Commit

6 years ago

Merge pull request #726 from google/dont-pack-all-outputs avoid packing leaf outputs for jit/pmap funs

References

Author

mattjj

Parents