Commit 14e6c78
Reimplement JAX Scan dispatcher with MIT-MOT support
Co-authored-by: Jesse Grabowski <[email protected]>1 parent 9779797 commit 14e6c78
File tree
3 files changed
+320
-185
lines changed- pytensor
- link/jax/dispatch
- scan
- tests/link/jax
3 files changed
+320
-185
lines changed
0 commit comments