Best practice for nonlinear, time-dependent PDE likelihood: scan(), custom Op with JAX, custom Op with FEniCS

Questions
2

FYI: If you are using custom Op with JAX, you will also need to wrap the Op in Aesara for PyMC to use it.

1 Like