Composable kernels for scikit-learn implemented in JAX.
☆47Oct 26, 2020Updated 6 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
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- JAX Version of VMEC2000☆16Updated this week
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Oct 5, 2021Updated 5 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- ☆658Oct 20, 2026Updated 2 weeks ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 5 years ago
- [ICLR'26] A scalable, effective, and general way to augment LLMs with billion-scale knowledge graphs using very little GPU memory cost.☆26Jan 27, 2026Updated 9 months ago
- Gravitational-wave data analysis tools in JAX☆16Updated 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
- Tutorial materials of the Probabilistic Numerics Spring School.☆37May 10, 2023Updated 3 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
- The official code for the Harmonic variational Gaussian processes (ICML2021).☆11Jun 6, 2021Updated 5 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆16Apr 26, 2023Updated 3 years ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆44Mar 1, 2025Updated last year
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 5 years ago
- Differentiable gravitational waveforms in JAX☆17Updated this week
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆25Updated this week
- 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
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- ☆13Aug 17, 2020Updated 6 years ago
- The Brain Predictability toolbox (BPt), is a python based Machine Learning library designed primarily for tabular and neuroimaging specif…☆11Dec 14, 2022Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Graph Learning with JAX☆14Jul 11, 2022Updated 4 years ago
- Neural Networks for JAX☆84Sep 24, 2024Updated 2 years ago
- Provides a heteroscedastic noise latent for a sparse variational Gaussian process using GPflow☆13Nov 11, 2020Updated 5 years ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆245Dec 22, 2023Updated 2 years ago
- Summer Scheming!!!!!!☆11Aug 20, 2020Updated 6 years ago
- General-purpose differentiable meshfree solver built on JAX and the Material Point Method (MPM), supporting GPU-accelerated forward and i…☆19Updated this week
- Code for NeurIPS 2021 paper 'Spatio-Temporal Variational Gaussian Processes'☆49Nov 11, 2021Updated 4 years ago
- Interactive textbook on state space models☆12Apr 11, 2022Updated 4 years ago
- Gaussian processes in JAX and Equinox.☆669Updated this week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- PIX is an image processing library in JAX, for JAX.☆452Updated this week
- Exponential families for JAX☆77Updated this week
- 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
- Code and data sets for the review on graph kernels☆11Feb 22, 2022Updated 4 years ago
- A Cython interface to FLANN☆24Nov 25, 2020Updated 5 years ago
- Kernels, the machine learning ones☆15Mar 6, 2023Updated 3 years ago
- ☆113Jun 25, 2024Updated 2 years ago