Extending JAX with custom C++ and CUDA code
Primary LanguagePythonMIT LicenseMIT
No one’s star this repository yet.