aradha / recursive_feature_machinesLinks
☆51Updated last month
Alternatives and similar repositories for recursive_feature_machines
Users that are interested in recursive_feature_machines are comparing it to the libraries listed below
Sorting:
- Code for verifying deep neural feature ansatz☆19Updated 2 years ago
- Riemannian Optimization Using JAX☆49Updated last year
- ☆13Updated last year
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated 2 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆58Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆21Updated 10 months ago
- ☆26Updated 2 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Implementation of approximate free-energy minimization in PyTorch☆19Updated 3 years ago
- Neural Tangent Kernel (NTK) module for the scikit-learn library☆25Updated 7 months ago
- Bayesian optimization with conformal coverage guarantees☆28Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆175Updated this week
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62Updated 4 years ago
- PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆37Updated 3 years ago
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆61Updated last year
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆20Updated 6 years ago
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- Pytorch implementation of VAEs for heterogeneous likelihoods.☆42Updated 2 years ago
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆10Updated 2 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- Normalizing Flows with a resampled base distribution☆47Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆38Updated last month
- Neat Bayesian machine learning examples☆58Updated 4 months ago
- Repo to accompany paper "Implicit Self-Regularization in Deep Neural Networks..."☆44Updated 6 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- ☆30Updated 2 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆105Updated 4 years ago