Last released Apr 17, 2026
Custom CUDA extension for GraphGP package.
Last released Mar 31, 2026
A simple JAX interpreter for models.
Last released Dec 4, 2025
Scalable Gaussian processes with nearest neighbor graphs.
Last released Jul 15, 2025
Minimal JAX implementation of k-nearest neighbors using a k-d tree.
Last released Jun 18, 2025
Extension package for jaxkd.
Supported by
