sklearn-jax-kernels
Composable kernels for scikit-learn implemented in JAX.
AI summary
Kernel library
A set of composable kernels for scikit-learn implemented in JAX to accelerate computation and gradient calculation.
- stars
- 42
- forks
- 4
- watching
- 5
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
ML library
A Julia implementation of popular machine learning algorithms and interfaces.
Kernel toolbox
A collection of MATLAB programs implementing kernel-based algorithms for nonlinear signal processing and machine learning.
RL library
Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces.
Transpiler
A tool for transpiling scikit-learn estimators to various programming languages.
ML framework
Enables machine learning with scikit-learn in OCaml
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
Stencil library
A JAX-based stencil computation library for efficient numerical computations
JAX tutorial series
A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning.
Machine Learning Tutorial
Tutorials and materials for learning machine learning with Python using popular libraries like scikit-learn.
parameter optimizer
Automates search for optimal parameters in machine learning algorithms.
Kernel factory
A factory for creating kernel functions used in machine learning algorithms
Machine Learning Dataset Library
A Racket module for managing machine learning datasets and providing interfaces to various classification algorithms.
Machine learning library
A Rust machine learning crate providing algorithms and utilities for building and training machine learning models.
KFAC optimizer
Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions sklearn-jax-kernels.
