Extending JAX with custom C++ and CUDA code
Primary LanguagePythonMIT LicenseMIT
No issues in this repository yet.