Composable kernels for scikit-learn implemented in JAX.
☆47Oct 26, 2020Updated 5 years ago
Alternatives and similar repositories for sklearn-jax-kernels
Users that are interested in sklearn-jax-kernels are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆103Jul 6, 2023Updated 3 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Feb 24, 2020Updated 6 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 5 years ago
- ☆657Updated this week
- Rudi, A., Camoriano, R. and Rosasco, L., Less is more: Nyström computational regularization. In Advances in Neural Information Processing…☆13Sep 17, 2019Updated 7 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆50Oct 22, 2020Updated 5 years ago
- Batch and incremental Sparse Spectrum Gaussian Process for Regression☆10Mar 16, 2021Updated 5 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆331Mar 20, 2024Updated 2 years ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- ☆16Apr 26, 2023Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 5 years ago
- Website for the book "The Elements of Differentiable Programming".☆15Jul 2, 2025Updated last year
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆22Jul 20, 2024Updated 2 years ago
- fork of rl-baseline-zoo☆21Jul 1, 2020Updated 6 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆243Dec 22, 2023Updated 2 years ago
- Provides a heteroscedastic noise latent for a sparse variational Gaussian process using GPflow☆13Nov 11, 2020Updated 5 years ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- ☆13Aug 17, 2020Updated 6 years ago
- Neural Networks for JAX☆84Sep 24, 2024Updated 2 years ago
- Graph Learning with JAX☆14Jul 11, 2022Updated 4 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- A repository with implementations of major papers on Gaussian Process regression models, implemented from scratch in Python, notably incl…☆14Nov 19, 2022Updated 3 years ago
- Exponential families for JAX☆77Oct 10, 2026Updated last week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47May 31, 2024Updated 2 years ago
- Code for NeurIPS 2021 paper 'Spatio-Temporal Variational Gaussian Processes'☆49Nov 11, 2021Updated 4 years ago
- Spatial transcriptomics foundation model based on Graph Transformer and JEPA☆74Aug 10, 2026Updated 2 months ago
- Summer Scheming!!!!!!☆11Aug 20, 2020Updated 6 years ago
- ☆113Jun 25, 2024Updated 2 years ago
- Interactive textbook on state space models☆12Apr 11, 2022Updated 4 years ago
- FALKON implementation used in the experimental section of "FALKON: An Optimal Large Scale Kernel Method"☆30Oct 15, 2020Updated 6 years ago
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 3 years ago
- The tiniest of Gaussian Process libraries☆350Oct 5, 2026Updated last week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A Julia implementation of sparse Gaussian processes via path-wise doubly stochastic variational inference.☆33Aug 27, 2020Updated 6 years ago
- Analogous Safe-state Exploration (ASE) is an algorithm for provably safe and optimal exploration in MDPs with unknown, stochastic dynamic…☆10Mar 22, 2021Updated 5 years ago
- Gaussian processes with spherical harmonic features in JAX☆16Aug 24, 2025Updated last year
- A High Level API for Deep Learning in JAX☆472Dec 15, 2022Updated 3 years ago
- ☆154May 25, 2020Updated 6 years ago
- Training methodologies for autoregressive neural operators/emulators in JAX.☆15Jun 1, 2026Updated 4 months ago
- Multi-agent AI system for drug-target identification and due diligence.☆133Sep 17, 2026Updated last month