sklearn-jax-kernels
Kernel library
A set of composable kernels for scikit-learn implemented in JAX to accelerate computation and gradient calculation.
Composable kernels for scikit-learn implemented in JAX.
42 stars
5 watching
4 forks
Language: Python
last commit: about 4 years ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 cstjean/scikitlearn.jl | A Julia implementation of popular machine learning algorithms and interfaces. | 544 |
 steven2358/kmbox | A collection of MATLAB programs implementing kernel-based algorithms for nonlinear signal processing and machine learning. | 52 |
 ikostrikov/jaxrl | Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces. | 630 |
 intel/scikit-learn-intelex | An acceleration toolkit for scikit-learn machine learning algorithms on Intel hardware | 1,227 |
 nok/sklearn-porter | A tool for transpiling scikit-learn estimators to various programming languages. | 1,293 |
 lehy/ocaml-sklearn | Enables machine learning with scikit-learn in OCaml | 34 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 asem000/kernex | A JAX-based stencil computation library for efficient numerical computations | 66 |
 gordicaleksa/get-started-with-jax | A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning. | 661 |
 amueller/scipy_2015_sklearn_tutorial | Tutorials and materials for learning machine learning with Python using popular libraries like scikit-learn. | 576 |
 hyperopt/hyperopt-sklearn | Automates search for optimal parameters in machine learning algorithms. | 1,588 |
 mljs/kernel | A factory for creating kernel functions used in machine learning algorithms | 1 |
 johnstonskj/rml-core | A Racket module for managing machine learning datasets and providing interfaces to various classification algorithms. | 13 |
 maciejkula/rustlearn | A Rust machine learning crate providing algorithms and utilities for building and training machine learning models. | 619 |
 google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 248 |