jax
985ecbe5 - Merge pull request #1235 from google/vmap-axes-lists

Commit

7 years ago

Merge pull request #1235 from google/vmap-axes-lists allow vmap in_axes to be lists

References

Author

mattjj

Parents