-
Notifications
You must be signed in to change notification settings - Fork 20
Open
Description
There's a minor typo at https://pymc-devs.github.io/nutpie/pymc-usage.html#choosing-the-backend
I think
compiled_jax = nutpie.compiled_pymc_model(model, backend="jax")
trace = nutpie.sample(compiled_jax)
should be
compiled_jax = nutpie.compile_pymc_model(model, backend="jax")
trace = nutpie.sample(compiled_jax)
I hope that helps. Thank you for the package! It's made my dissertation, and by extension my life, much easier!
maresb
Metadata
Metadata
Assignees
Labels
No labels