cvxgrp / torch_linops
A library to define abstract linear operators, and associated algebra and matrix-free algorithms, that works with pyTorch Tensors.
☆12Updated 3 months ago
Alternatives and similar repositories for torch_linops
Users that are interested in torch_linops are comparing it to the libraries listed below
Sorting:
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆57Updated 5 months ago
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆57Updated last year
- Anderson accelerated Douglas-Rachford splitting☆29Updated 4 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆28Updated 6 months ago
- Proximal algorithms for nonsmooth optimization in Julia☆133Updated 4 months ago
- ☆26Updated 2 weeks ago
- Solve sparse linear systems in JAX using the KLU algorithm☆35Updated this week
- ☆47Updated 2 weeks ago
- APPM 5630 at CU Boulder☆48Updated 2 weeks ago
- Differentiable interface to FEniCS for JAX☆53Updated 3 years ago
- Code for ResDMD: data-driven spectral properties of Koopman Operators☆38Updated last year
- Riemannian Interior Point Methods (RIPM) for Constrained Optimization on Manifolds☆13Updated last year
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Updated 9 months ago
- Differentiation through cone programs☆99Updated 3 months ago
- Python Algorithms for Randomized Linear Algebra☆53Updated 2 years ago
- Constrained optimization for Pytorch using the SQP-GS algorithm☆33Updated 11 months ago
- Proximal operators for nonsmooth optimization in Julia☆134Updated last year
- Python Tensor Toolbox☆33Updated 3 weeks ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆25Updated last year
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆15Updated last year
- Mathematical Programming in JAX☆76Updated last week
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆48Updated 8 months ago
- A matlab implementation of forward automatic differentiation with operator overloading and sparse jacobians☆43Updated 2 months ago
- ☆13Updated 8 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated 2 months ago
- Numerical quadrature with JAX☆58Updated last week
- Globally Convergent Type-I Anderson Acceleration for Non-Smooth Fixed-Point Iterations☆20Updated 6 years ago
- Matrix-free linear algebra in JAX.☆118Updated 4 months ago
- Example code for paper: Automatic Differentiation to Simultaneously Identify Nonlinear Dynamics and Extract Noise Probability Distributio…☆55Updated 2 years ago
- ☆51Updated 7 months ago