PDMPs in JAX
Primary LanguagePythonApache License 2.0Apache-2.0
PDMP based inference in JAX.
Take a look at the examples in the examples folder.