Code

Fix Scan JAX dispatcher#232

Code

Merged

ricardoV94 merged 4 commits intopymc-devs:mainpymc-devs/pytensor:mainfrom

ricardoV94:fix_jax_scan_dispatchricardoV94/pytensor:fix_jax_scan_dispatchCopy head branch name to clipboard

Apr 4, 2023

Commits

Commits on Apr 4, 2023

• Allow overriding modes used in `compare_jax_and_py` helper

  ricardoV94

  committed
  05d9199

  Copy full SHA for 05d9199

• Return PyTensor JAX function in `compare_jax_and_py` helper

  ricardoV94

  committed
  cffd161

  Copy full SHA for cffd161

• Use `"JAX"` mode by default when testing jax dispatching

  ricardoV94

  committed
  4e29318

  Copy full SHA for 4e29318

• Fix Scan JAX dispatcher

  ricardoV94

  committed
  1ba5887

  Copy full SHA for 1ba5887