/extending-jax

Extending JAX with custom C++ and CUDA code

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.