andres-fr / skerchLinks
Sketched linear operations for PyTorch
☆100Updated 2 months ago
Alternatives and similar repositories for skerch
Users that are interested in skerch are comparing it to the libraries listed below
Sorting:
- Algorithms for computations on random manifolds made easier☆94Updated 2 years ago
- IVON optimizer for neural networks based on variational learning.☆80Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆62Updated this week
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆151Updated 2 months ago
- Agustinus' very opiniated publication-ready plotting library☆70Updated 8 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Updated last month
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Source code for my PhD thesis: Backpropagation Beyond the Gradient☆20Updated 2 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆191Updated last year
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated 2 years ago
- Normalizing Flows using JAX☆86Updated 2 years ago
- ☆52Updated 2 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆108Updated last year
- Compositional Linear Algebra☆505Updated 5 months ago
- Stochastic Automatic Differentiation library for PyTorch.☆208Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆191Updated last week
- Matrix-free linear algebra in JAX.☆160Updated last month
- A generic interface for linear algebra backends☆76Updated 2 months ago
- Laplace approximations in JAX.☆41Updated this week
- Convex potential flows☆85Updated 4 years ago
- JAX Arrays for human consumption☆110Updated 2 months ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 10 months ago
- Run PyTorch in JAX. 🤝☆311Updated 3 months ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆171Updated 3 years ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated last year
- Simulation-based inference benchmark☆108Updated 11 months ago