Theano Op using JAX for lightning-fast ODE inference

twiecki April 15, 2021, 3:31pm 3

This is really cool - thanks for sharing. Would be a great contribution to PyMC3!

1 Like

Topic		Replies	Views
ODEs, PyMC4 and custom likelihood in jax Questions modeling	9	1616	April 10, 2022
Best practice for nonlinear, time-dependent PDE likelihood: scan(), custom Op with JAX, custom Op with FEniCS Questions ode_pde	2	873	August 23, 2022
How to write a PyTensor Op to wrap Jax ODEs with multiple input parameters v5 jax	7	984	July 28, 2024
How to use JAX ODEs and Neural Networks in PyMC Sharing jax	0	437	January 4, 2023
ODE with matrix calculation v3	3	468	April 25, 2022