dfm / extending-jaxLinks

Extending JAX with custom C++ and CUDA code
394Updated 9 months ago

Alternatives and similar repositories for extending-jax

Users that are interested in extending-jax are comparing it to the libraries listed below

Sorting: