flaport / torch_sparse_solve
A sparse KLU solver for PyTorch.
☆62Updated 3 years ago
Alternatives and similar repositories for torch_sparse_solve:
Users that are interested in torch_sparse_solve are comparing it to the libraries listed below
- Preconditioned Conjugate Gradient in Pytorch☆140Updated last year
- Solve sparse linear systems in JAX using the KLU algorithm☆34Updated 2 weeks ago
- Convex potential flows☆83Updated 3 years ago
- PyTorch implementation of the Hessian-free optimizer☆34Updated 9 months ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆93Updated last year
- Interpolation routines in Pytorch.☆110Updated 4 years ago
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆93Updated 3 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆70Updated 10 months ago
- Python wrapper for the sparse QR decomposition in SuiteSparseQR.☆35Updated last month
- ☆101Updated 4 years ago
- Implicit^2: Implicit model for implicit neural representations☆28Updated 3 years ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 3 years ago
- ☆56Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆57Updated 3 years ago
- Source code for Deep Multigrid method https://arxiv.org/pdf/1711.03825.pdf☆18Updated 5 years ago
- ☆22Updated 2 years ago
- ☆11Updated 3 years ago
- Anderson accelerated Douglas-Rachford splitting☆29Updated 4 years ago
- ☆63Updated last year
- Scale-invariant Learning by Physics Inversion (NeurIPS 2022)☆10Updated 2 years ago
- Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.☆42Updated 4 years ago
- PyTorch-FEniCS interface☆100Updated 3 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆55Updated 3 months ago
- Riemannian Convex Potential Maps☆67Updated last year
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆59Updated 4 months ago
- Turning SymPy expressions into PyTorch modules.☆146Updated last year
- Sparse matrix tools extending scipy.sparse, but with incompatible licenses☆165Updated 5 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 6 months ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated last year