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	766	October 5, 2021
ODEs, PyMC4 and custom likelihood in jax Questions modeling	9	1509	April 10, 2022
Creating Theano Op to work on vectors element-wise within Model()? Questions	5	2440	August 30, 2017
New package: Transforming JAX to Pytensor, for ODEs and other applications Development	2	80	November 28, 2024
Jax (and leapfrog integration) for differential equation approximate solutions? Questions	1	724	April 18, 2021