aradha / recursive_feature_machines
☆36Updated last week
Related projects: ⓘ
- Code for verifying deep neural feature ansatz☆14Updated last year
- Implementation of approximate free-energy minimization in PyTorch☆17Updated 2 years ago
- ☆29Updated last year
- Graph Learning with JAX☆14Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆43Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆41Updated 3 years ago
- Bayesian optimization with conformal coverage guarantees☆26Updated last year
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆113Updated last year
- TARDIS: Topological Algorithms for Robust DIscovery of Singularities☆39Updated last year
- Laplace Redux -- Effortless Bayesian Deep Learning☆35Updated last year
- ☆49Updated last year
- Quantification of Uncertainty with Adversarial Models☆27Updated last year
- Neat Bayesian machine learning examples☆52Updated last week
- Riemannian Optimization Using JAX☆44Updated 10 months ago
- Bayesian inference with Python and Jax.☆31Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆34Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆94Updated 10 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆68Updated last year
- Bayesian algorithm execution (BAX)☆46Updated 3 years ago
- Neural Graphical models are neural network based graphical models that offer richer representation, faster inference & sampling☆24Updated 10 months ago
- Random feature latent variable models in Python☆21Updated last year
- Public code for running Stochastic Gradient Descent on GPs.☆31Updated 5 months ago
- Implementation of Action Matching☆35Updated last year
- A Python package for intrinsic dimension estimation☆78Updated last month
- ☆30Updated 3 months ago
- Pytorch implementation of VAEs for heterogeneous likelihoods.☆41Updated last year
- "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"☆16Updated 11 months ago
- Treeffuser is an easy-to-use package for probabilistic prediction on tabular data with tree-based diffusion models.☆11Updated last month
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆115Updated 3 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆23Updated last year