flaport / torch_sparse_solve
A sparse KLU solver for PyTorch.
☆65Updated 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.☆110Updated 4 years ago
- Convex potential flows☆83Updated 3 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆35Updated this week
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago
- ☆101Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆95Updated last year
- PyTorch implementation of the Hessian-free optimizer☆34Updated 11 months ago
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆94Updated 4 years ago
- Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers☆157Updated 2 years ago
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆72Updated 9 months ago
- A wrapper around scipy minimize which uses the autograd capacities of tensorflow to compute the gradient and hessian.☆27Updated last year
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆55Updated 4 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆57Updated 5 months ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Code for the ICLR 2020 paper "Learning to Control PDEs"☆32Updated 5 years ago
- PyTorch-FEniCS interface☆100Updated 4 years ago
- ☆22Updated 4 years ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆63Updated 6 months ago
- The differentiable OSQP solver layer for PyTorch☆59Updated 2 years ago
- Path Tracing in JAX☆71Updated 5 years ago
- MintNet: Building Invertible Neural Networks with Masked Convolutions☆39Updated 4 years ago
- Newton and Quasi-Newton optimization with PyTorch☆345Updated last year
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆18Updated last year
- Auto-differentiable and hardware-accelerated force density method☆88Updated 3 months ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆63Updated 3 months ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated 11 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Python wrapper for the sparse QR decomposition in SuiteSparseQR.☆35Updated 3 months ago
- Sparse matrix tools extending scipy.sparse, but with incompatible licenses☆168Updated 2 weeks ago