jax
e06a6ab6 - Add support for negative axes to vmap. (#4111)

Commit

5 years ago

Add support for negative axes to vmap. (#4111) * Add support for negative axes to vmap. * Add workaround for out-of-range vmap axes.

References

Author

hawkinsp

Parents