sphinxteam / tramp
Tree Approximate Message Passing
☆30Updated 9 months ago
Alternatives and similar repositories for tramp:
Users that are interested in tramp are comparing it to the libraries listed below
- ☆25Updated 6 years ago
- ☆22Updated 4 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- Efficient Householder Transformation in PyTorch☆64Updated 3 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆72Updated 6 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆48Updated 5 years ago
- ☆47Updated 5 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 5 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Differentiable and numerically stable implementation of the matrix exponential☆32Updated 4 years ago
- Optimization with orthogonal constraints and on general manifolds☆127Updated 4 years ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated 7 months ago
- ☆69Updated last year
- Autoregressive Energy Machines☆77Updated 2 years ago
- Reliable Uncertainty Estimates in Deep Neural Networks using Noise Contrastive Priors☆63Updated 4 years ago
- ☆37Updated 5 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Convex potential flows☆82Updated 3 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 6 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆13Updated 5 years ago
- ☆31Updated 4 years ago
- ☆80Updated 3 years ago
- Code for "Accelerating Natural Gradient with Higher-Order Invariance"☆30Updated 5 years ago
- ☆13Updated 6 months ago
- Code for "Differentiable Compositional Kernel Learning for Gaussian Processes" https://arxiv.org/abs/1806.04326☆71Updated 6 years ago
- Examples of more involved applications using Geomstats☆31Updated 3 years ago