jax
71d5a7ab - Merge pull request #1336 from google/fix-vmap-of-pmap

Commit

6 years ago

Merge pull request #1336 from google/fix-vmap-of-pmap fix vmap-of-pmap bug

References

#1336 - fix vmap-of-pmap bug

Author

mattjj

mattjj

Parents

Loading