Skip to content

HGangloff/jax_ffi_experiments

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
array_split
array_split
 
 
gibbs_sampler
gibbs_sampler
 
 
.gitignore
.gitignore
 
 
README.md
README.md
 
 

Repository files navigation

Two examples of usage for JAX FFI interface:

  • A well detailed C++ vanilla Gibbs sampler implementation called from JAX
  • A jnp.split C++ reimplementation. Includes a use example for RemainingRets.

About

JAX Foreign Function Interface experiments

Topics

tutorial jax xla foreign-function-interface

Resources

Readme
Activity

Stars

0 stars

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages