cvxgrp / torch_linops
A library to define abstract linear operators, and associated algebra and matrix-free algorithms, that works with pyTorch Tensors.
☆12Updated last month
Alternatives and similar repositories for torch_linops:
Users that are interested in torch_linops are comparing it to the libraries listed below
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆55Updated 3 months ago
- Anderson accelerated Douglas-Rachford splitting☆29Updated 4 years ago
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆56Updated last year
- Riemannian Interior Point Methods (RIPM) for Constrained Optimization on Manifolds☆13Updated last year
- Proximal algorithms for nonsmooth optimization in Julia☆133Updated 2 months ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆25Updated 10 months ago
- APPM 5630 at CU Boulder☆46Updated this week
- ☆19Updated last year
- ☆17Updated this week
- Solve sparse linear systems in JAX using the KLU algorithm☆34Updated last week
- Riemannian optimization on the symplectic Stiefel manifold☆13Updated 2 years ago
- A sparse polynomial optimization tool based on the moment-SOS hierarchy.☆50Updated this week
- Proximal operators for nonsmooth optimization in Julia☆134Updated last year
- Python Algorithms for Randomized Linear Algebra☆53Updated last year
- Differentiable interface to FEniCS for JAX☆53Updated 3 years ago
- NonOpt: Nonlinear/nonconvex/nonsmooth Optimizer☆10Updated 2 weeks ago
- Turning SymPy expressions into JAX functions☆44Updated 3 years ago
- Differentiation through cone programs☆96Updated last month
- ☆51Updated 5 months ago
- ☆48Updated this week
- ☆12Updated 3 years ago
- Mathematical Programming in JAX☆69Updated last week
- Efficient Differentiable n-d PDE solvers in JAX.☆24Updated 4 months ago
- Functional models and algorithms for sparse signal processing☆89Updated last year
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆41Updated last week
- ☆12Updated 6 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆105Updated last month
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Updated 7 months ago
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆14Updated last year
- Fast hyperparameter settings for non-smooth estimators:☆39Updated last year