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
Sorting:
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆103Jul 6, 2023Updated 2 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆330Mar 20, 2024Updated last year
- ☆620Jan 23, 2026Updated last month
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47May 31, 2024Updated last year
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Summer Scheming!!!!!!☆11Aug 20, 2020Updated 5 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Feb 24, 2020Updated 6 years ago
- Low-rank Highway Networks☆13Mar 11, 2016Updated 9 years ago
- fork of rl-baseline-zoo☆21Jul 1, 2020Updated 5 years ago
- Exponential families for JAX☆76Feb 20, 2026Updated last week
- ☆50Oct 22, 2020Updated 5 years ago
- ☆13Aug 17, 2020Updated 5 years ago
- Kernels, the machine learning ones☆14Mar 6, 2023Updated 2 years ago
- Provides a heteroscedastic noise latent for a sparse variational Gaussian process using GPflow☆13Nov 11, 2020Updated 5 years ago
- Rudi, A., Camoriano, R. and Rosasco, L., Less is more: Nyström computational regularization. In Advances in Neural Information Processing…☆13Sep 17, 2019Updated 6 years ago
- Interactive textbook on state space models☆12Apr 11, 2022Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 4 years ago
- Recommendation engine for scholarly articles☆12Oct 22, 2019Updated 6 years ago
- Python implementation of tabular asynchronous actor critic☆11May 3, 2016Updated 9 years ago
- Imputation of single cell RNA-sequencing data with autoencoder☆16Jun 23, 2021Updated 4 years ago
- Unofficial Experiments with AlgebraNets☆17Jun 17, 2020Updated 5 years ago
- Maybe next gen of Pyzo IDE based on Flexx☆17Dec 17, 2017Updated 8 years ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- ☆15Jan 15, 2019Updated 7 years ago
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 2 years ago
- SCIENTIfic coLLaborative Archive☆11Aug 3, 2015Updated 10 years ago
- An opinionated template for spinning up a dask cluster based on docker.☆14Nov 13, 2017Updated 8 years ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Dec 20, 2023Updated 2 years ago
- This is a short introduction of Julia language. This is in English and Japanese.☆14Nov 6, 2024Updated last year
- Code & Experiments for "LILA: Language-Informed Latent Actions" to be presented at the Conference on Robot Learning (CoRL) 2021.☆14Nov 4, 2021Updated 4 years ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated 11 months ago
- Gaussian processes in JAX and Flax.☆593Feb 20, 2026Updated last week
- 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
- Python binding of primitiv.☆17Sep 12, 2022Updated 3 years ago
- An implementation of the differentially private variational inference algorithm for NumPyro.☆16Sep 3, 2024Updated last year
- Implicit generative models and related stuff based on the MMD, in PyTorch☆16Sep 24, 2020Updated 5 years ago
- Graph Learning with JAX☆14Jul 11, 2022Updated 3 years ago
- Makes a bunch of EC2 spot priced instances and starts dask running on them.☆13Jun 18, 2018Updated 7 years ago
- Gated Recurrent Unit with Low-rank matrix factorization☆34Mar 11, 2016Updated 9 years ago