Skip to content

Compile the function ahead of time in the JAX example#6286

Open
rostan-t wants to merge 1 commit intoNVIDIA:mainfrom
rostan-t:jax-example-fix-graph-capture
Open

Compile the function ahead of time in the JAX example#6286
rostan-t wants to merge 1 commit intoNVIDIA:mainfrom
rostan-t:jax-example-fix-graph-capture

Commits

Commits on Apr 9, 2026