jax
d158647c - Merge pull request #4706 from apaszke:vmap-collectives-in-scan

Commit

5 years ago

Merge pull request #4706 from apaszke:vmap-collectives-in-scan PiperOrigin-RevId: 339646941

References

Author

a-googler

Parents