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
Fast ODE solver for PyMC3 Questions	3	775	October 5, 2021
New package: Transforming JAX to Pytensor, for ODEs and other applications Development	3	161	March 19, 2025
ODEs, PyMC4 and custom likelihood in jax Questions modeling	9	1520	April 10, 2022
Creating Theano Op to work on vectors element-wise within Model()? Questions	5	2445	August 30, 2017
Jax (and leapfrog integration) for differential equation approximate solutions? Questions	1	737	April 18, 2021