tests/link/jax/test_random.py · bb028ae2330433755b9d4aa32ab6e8d0c9f662fc · testgroup / pytensor · GitLab

Blame History Permalink

Inline static size inputs in JAX implementation of RandomVariables · bb028ae2
由 Ricardo Vieira 提交于 5月 17, 2024
```
This gets around some limitations in JAX jitting system
```
bb028ae2

test_random.py 28.9 KB