jax
ebbcbad5 - allow vmap in_axes to be a list, fixes #2367 (#2395)

Commit

5 years ago

allow vmap in_axes to be a list, fixes #2367 (#2395)

References

#2395 - allow vmap in_axes to be a list, fixes #2367

Author

mattjj

mattjj

Parents

Loading