jax
3815662a - Merge pull request #755 from google/scan-vmap

Commit

6 years ago

Merge pull request #755 from google/scan-vmap add vmap support to `lax.scan` (so vmap-of-scan works)

References

Author

mattjj

Parents