Two examples of usage for JAX FFI interface:
- A well detailed C++ vanilla Gibbs sampler implementation called from JAX
- A
jnp.splitC++ reimplementation. Includes a use example forRemainingRets.
HGangloff/jax_ffi_experiments
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|