jax
051abc1b - Merge pull request #816 from google/vmap-of-pmap

Commit

6 years ago

Merge pull request #816 from google/vmap-of-pmap add vmap-of-pmap

References

#816 - add vmap-of-pmap

Author

mattjj

mattjj

Parents

Loading