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