flaport / torch_sparse_solveLinks
A sparse KLU solver for PyTorch.
☆66Updated 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
Sorting:
- Preconditioned Conjugate Gradient in Pytorch☆140Updated last year
- Interpolation routines in Pytorch.☆111Updated 4 years ago
- Neural reparameterization improves structural optimization☆125Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆102Updated last year
- ☆103Updated 4 years ago
- Convex potential flows☆84Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 2 years ago
- Turning SymPy expressions into PyTorch modules.☆151Updated 2 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated last year
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆63Updated 11 months ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 4 years ago
- Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers☆164Updated 2 years ago
- Auto-differentiable and hardware-accelerated force density method☆93Updated 2 months ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆267Updated 3 years ago
- Orthogonal polynomials in all shapes and sizes.☆188Updated last year
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆121Updated 2 years ago
- Differentiable interface to FEniCS for JAX☆58Updated 4 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- Python implementation of the paper "Discrete Differential-Geometry Operators for Triangulated 2-Manifolds" by Meyer et. al. VisMath 2002☆53Updated 2 years ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆56Updated 4 years ago
- Path Tracing in JAX☆71Updated 5 years ago
- Implicit^2: Implicit model for implicit neural representations☆29Updated 3 years ago
- PyTorch-FEniCS interface☆102Updated 4 years ago
- A library of Krylov methods in pure Python☆63Updated 2 years ago
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆26Updated 4 years ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated last year
- Sparse matrix tools extending scipy.sparse, but with incompatible licenses☆177Updated 5 months ago
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆59Updated 4 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆36Updated 4 months ago
- Source code for the ICLR'22 paper on "Half-Inverse Gradients"☆18Updated 3 years ago