jax
d48e7ef4 - add batching (vmap) rule for lax.dynamic_slice

Commit

7 years ago

add batching (vmap) rule for lax.dynamic_slice fixes #165

References

#168 - add batching (vmap) rule for lax.dynamic_slice

Author

mattjj

mattjj

Parents

Loading