feat: support simple sequential batching rule (support for jax.vmap and jax.jacobian)#47
Merged
jpbrodrick89 merged 14 commits intomainfrom Jun 23, 2025
Merged
feat: support simple sequential batching rule (support for `jax.vmap` and `jax.jacobian`)#47jpbrodrick89 merged 14 commits intomainfrom
jpbrodrick89 merged 14 commits intomainfrom
Commits
Commits on Jun 20, 2025
- committed
- committed
- authored
- committed
- committed
Commits on Jun 21, 2025
Commits on Jun 23, 2025
- committed
- committed
- committed
- committed
- committed
- committed
- andauthored
