jax
4fe2bcfe - add vmap-of-scan tests, fix minor bugs

Commit

6 years ago

add vmap-of-scan tests, fix minor bugs

References

#755 - add vmap support to `lax.scan` (so vmap-of-scan works)

Author

mattjj

mattjj

Committer

mattjj

mattjj

Parents

Loading