jax
fd3645ab - Merge pull request #168 from google/vmap-for-dynamic-slice

Commit

7 years ago

Merge pull request #168 from google/vmap-for-dynamic-slice add batching (vmap) rule for lax.dynamic_slice

References

Author

mattjj

Parents