Composable kernels for scikit-learn implemented in JAX.
☆46Oct 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:
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Oct 5, 2021Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Feb 24, 2020Updated 6 years ago
- Batch and incremental Sparse Spectrum Gaussian Process for Regression☆10Mar 16, 2021Updated 5 years ago
- ☆16Apr 26, 2023Updated 3 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆331Mar 20, 2024Updated 2 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 5 years ago
- Kernels, the machine learning ones☆14Mar 6, 2023Updated 3 years ago
- ☆50Oct 22, 2020Updated 5 years ago
- A fast, clean, responsive Hugo theme, now for academics.☆10Jun 20, 2025Updated 10 months ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- fork of rl-baseline-zoo☆21Jul 1, 2020Updated 5 years ago
- ☆634Apr 18, 2026Updated 2 weeks ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47May 31, 2024Updated last year
- Summer Scheming!!!!!!☆11Aug 20, 2020Updated 5 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Interactive textbook on state space models☆12Apr 11, 2022Updated 4 years ago
- Simple integration of keras-tuner (hyperparameter tuning) and tensorboard dashboard (interactive visualization).☆10Nov 24, 2020Updated 5 years ago
- A Python port of the R implementation of Kleinberg's burst detection algorithm☆12Apr 11, 2022Updated 4 years ago
- Mathematical operations for JAX pytrees☆210Dec 5, 2024Updated last year
- Gaussian processes with spherical harmonic features in JAX☆16Aug 24, 2025Updated 8 months ago
- Fully nonstationary, heteroscedastic GP for Matlab☆16Jul 25, 2024Updated last year
- Berkeley Single Cell Computational Microscopy dataset☆19Oct 27, 2025Updated 6 months 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
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆15Updated this 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.
- Material for dataviz course (Nancy, 2023)☆27Oct 27, 2023Updated 2 years ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- Code for the paper "Bayesian Neural Network Priors Revisited"☆61Jul 1, 2021Updated 4 years ago
- PIX is an image processing library in JAX, for JAX.☆436Mar 6, 2025Updated last year
- FALKON implementation used in the experimental section of "FALKON: An Optimal Large Scale Kernel Method"☆30Oct 15, 2020Updated 5 years ago
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 3 years ago
- Code & Experiments for "LILA: Language-Informed Latent Actions" to be presented at the Conference on Robot Learning (CoRL) 2021.☆13Nov 4, 2021Updated 4 years ago
- SCIENTIfic coLLaborative Archive☆11Aug 3, 2015Updated 10 years ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Dec 20, 2023Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆28Jan 12, 2022Updated 4 years ago
- Implicit generative models and related stuff based on the MMD, in PyTorch☆16Sep 24, 2020Updated 5 years ago
- Gaussian processes in JAX and Equinox.☆618Updated this week
- SMASHED is a toolkit designed to apply transformations to samples in datasets, such as fields extraction, tokenization, prompting, batchi…☆35May 24, 2024Updated last year
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆242Dec 22, 2023Updated 2 years ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Jul 20, 2024Updated last year
- ☆110Jun 25, 2024Updated last year