A lightweight didactic library of kernel methods using the back-end JAX.
☆12Mar 8, 2023Updated 3 years ago
Alternatives and similar repositories for jaxkern
Users that are interested in jaxkern are comparing it to the libraries listed below
Sorting:
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Dec 20, 2023Updated 2 years ago
- ☆18Dec 31, 2018Updated 7 years ago
- Fully nonstationary, heteroscedastic GP for Matlab☆16Jul 25, 2024Updated last year
- Non-stationary spectral mixture kernels implemented in GPflow☆29Nov 30, 2018Updated 7 years ago
- A conda-smithy repository for nvcc.☆13Jan 23, 2025Updated last year
- Code for the causal benchmark library☆12Mar 11, 2025Updated last year
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 years ago
- Online variational GPs☆38Jun 1, 2023Updated 2 years ago
- PyTorch implementation of GMLS-Nets. Machine learning methods for scattered unstructured data sets. Methods for learning differential op…☆27Nov 14, 2025Updated 4 months ago
- XAI Library☆20Nov 26, 2025Updated 3 months ago
- Gaussian processes in JAX and Flax.☆602Updated this week
- Vacances scolaires en France☆16Nov 14, 2025Updated 4 months ago
- ☆21Dec 8, 2022Updated 3 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 4 years ago
- Repo for the Tutorials of Day1-Day2 of the Nordic Probabilistic AI School 2023☆17Jun 13, 2023Updated 2 years ago
- ☆38Jun 3, 2024Updated last year
- FedRay: a Research Framework for Federated Learning based on Ray☆22Nov 20, 2024Updated last year
- A collection of useful, free, single-file libraries for C.☆11Oct 15, 2015Updated 10 years ago
- Kinematic and dynamic models of continuum and articulated soft robots.☆16Nov 22, 2025Updated 3 months ago
- Basic Gaussian process regression library. (Eigen3 required)☆27Jul 31, 2015Updated 10 years ago
- MMD-FUSE package implementing the MMD-FUSE test proposed in MMD-FUSE: Learning and Combining Kernels for Two-Sample Testing Without Data …☆11May 31, 2024Updated last year
- Code to support the guide to logical induction for software engineers☆11Mar 24, 2025Updated 11 months ago
- Particle Gibbs sampler for Bayesian additive regression trees (BART) in Rust.☆14Feb 6, 2026Updated last month
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Feb 20, 2023Updated 3 years ago
- Code supporting Bazinet et al., 2023, "Assortative mixing in micro-architecturally annotated brain connectomes" (Nature Communications)☆13May 19, 2023Updated 2 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Active Learning of Abstract Plan Feasibility☆12Feb 10, 2023Updated 3 years ago
- pylambertw - sklearn interface to analyze and gaussianize heavy-tailed, skewed data☆16Apr 15, 2024Updated last year
- Batch and incremental Sparse Spectrum Gaussian Process for Regression☆10Mar 16, 2021Updated 5 years ago
- Reproducibility code for MMD Aggregated Two-Sample Test, by Schrab, Kim, Albert, Laurent, Guedj and Gretton: https://arxiv.org/abs/2110.1…☆17May 31, 2024Updated last year
- A general-purpose lightweight sandbox for safely executing user programs☆16Jun 24, 2020Updated 5 years ago
- Uncertainty sets for nonlinear dynamical systems☆10Nov 7, 2020Updated 5 years ago
- A Gaussian process toolbox in python☆42Apr 20, 2012Updated 13 years ago
- A general purpose Gaussian process regression module☆11Jul 17, 2025Updated 8 months ago
- Code for the ICML 2019 paper: Distribution Calibration for Regression☆22Aug 8, 2023Updated 2 years ago
- Code for Deep Structured Mixtures of Gaussian Processes (DSMGPs)☆11Jan 27, 2022Updated 4 years ago
- Implementation of Self-supervised-Online-Adversarial-Purification☆13Aug 2, 2021Updated 4 years ago
- JAX implementation of Graph Attention Networks☆13Jan 29, 2022Updated 4 years ago
- 🌾 Synonyms, antonyms, rhymes & related words — in your terminal, Raycast, and web. Powered by Thesaurus.com & Datamuse.☆38Feb 15, 2026Updated last month