This repository was archived by the owner on Nov 17, 2025. It is now read-only.

Fix the JAX `Subtensor` and `IncSubtensor` dispatcher - #1338

Merged

rlouf merged 14 commits into

aesara-devs:mainfrom

rlouf:jax-fix-subtensor

Dec 14, 2022

Commits on Dec 14, 2022

Raise when an array is resized with a boolean mask
rlouf
committed
6591792
Use constants or raise in JAX `Arange` implementation
rlouf
committed
72930c9
Raise when trying to slice with a dynamic length
rlouf
committed
e099198
Use `jax.numpy.copy` directly
rlouf
committed
c01a0f3
Move `test_subtensor` back to the test suite
rlouf
committed
173e49a
Document discrepancy with `Clip` and `jax.numpy.clip`
rlouf
committed
7fb2ba1
Dispatch some `Op`s to Python operators when scalar inputs
rlouf
committed
8181685
Implement `TensorFromScalar` as a pass-through
rlouf
committed
3ddde94
Simplify the `IncSubtensor` dispatcher
rlouf
committed
60fbdad
Add rewrites to re-express boolean indexing logic
rlouf
committed
f9159e7
Typify 0-dim arrays to corresponding number
rlouf
committed
9cc0c28
Refactor the JAX implementation of `Reshape`
rlouf
committed
0202f6b
Fix the JAX implementation of `SpecifyShape`
rlouf
committed
e5f959e
Use separate DB queries for each JAX test mode

brandonwillard
authored and
rlouf
committed
78d2dbf