andres-fr / skerch
Sketched matrix decompositions for PyTorch
☆70Updated 2 weeks ago
Alternatives and similar repositories for skerch:
Users that are interested in skerch are comparing it to the libraries listed below
- Agustinus' very opiniated publication-ready plotting library☆62Updated 2 months ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆34Updated 3 weeks ago
- Algorithms for computations on random manifolds made easier☆89Updated last year
- IVON optimizer for neural networks based on variational learning.☆60Updated 4 months ago
- This repository contains code for applying Riemannian geometry in machine learning.☆77Updated 3 years ago
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆112Updated this week
- Tutorial materials of the Probabilistic Numerics Spring School.☆34Updated last year
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Squared Non-monotonic Probabilistic Circuits☆22Updated 2 months ago
- Source code for my PhD thesis: Backpropagation Beyond the Gradient☆20Updated 2 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 11 months ago
- Parameter-Free Optimizers for Pytorch☆122Updated 11 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- ☆50Updated 2 years ago
- Normalizing Flows using JAX☆83Updated last year
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆98Updated last year
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆21Updated 4 months ago
- Extensible Tensorflow library for differentiable particle filtering. ICML 2021.☆41Updated 2 years ago
- Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)☆13Updated last week
- Hessian spectral density estimation in TF and Jax☆122Updated 4 years ago
- ☆53Updated 8 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated 3 weeks ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆43Updated last year
- A generic interface for linear algebra backends☆73Updated last month
- Euclidean Wasserstein-2 optimal transportation☆46Updated last year
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆20Updated 5 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago