flaport / torch_sparse_solve
A sparse KLU solver for PyTorch.
☆64Updated 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
- PyTorch implementation of the Hessian-free optimizer☆34Updated 10 months ago
- Interpolation routines in Pytorch.☆110Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆94Updated last year
- Convex potential flows☆83Updated 3 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆35Updated last month
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆55Updated 4 months ago
- ☆101Updated 4 years ago
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆94Updated 3 years ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆62Updated 5 months ago
- ☆56Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago
- Newton and Quasi-Newton optimization with PyTorch☆342Updated last year
- Python wrapper for the sparse QR decomposition in SuiteSparseQR.☆35Updated 2 months ago
- Neural reparameterization improves structural optimization☆121Updated 3 years ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated last year
- Path Tracing in JAX☆71Updated 5 years ago
- Code for the ICLR 2020 paper "Learning to Control PDEs"☆32Updated 5 years ago
- The differentiable OSQP solver layer for PyTorch☆59Updated 2 years ago
- Differentiation through cone programs☆99Updated 2 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated last month
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆125Updated 7 months ago
- Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.☆42Updated 4 years ago
- Source code for the ICLR'22 paper on "Half-Inverse Gradients"☆18Updated 3 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆54Updated 4 years ago
- Turning SymPy expressions into PyTorch modules.☆148Updated last year
- Geometry processing utilities compatible with jax for autodifferentiation.☆88Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆61Updated 3 months ago
- B-Spline CNNs on Lie groups☆54Updated 5 years ago