Reinforcement learning - help building a model

Yes that’s unfortunately expected. Scan is not working in our JAX backend at the moment: JAX backend fails with simple Scan examples · Issue #924 · aesara-devs/aesara · GitHub

1 Like