flaport / torch_sparse_solve
A sparse KLU solver for PyTorch.
☆60Updated 2 years ago
Related projects: ⓘ
- Preconditioned Conjugate Gradient in Pytorch☆129Updated 7 months ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆87Updated 10 months ago
- Interpolation routines in Pytorch.☆107Updated 3 years ago
- Convex potential flows☆78Updated 2 years ago
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆90Updated 3 years ago
- PyTorch implementation of the Hessian-free optimizer☆30Updated 3 months ago
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆50Updated 3 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆24Updated 4 months ago
- Code for the ICLR 2020 paper "Learning to Control PDEs"☆29Updated 4 years ago
- Neural reparameterization improves structural optimization☆117Updated 3 years ago
- ☆98Updated 3 years ago
- Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.☆42Updated 4 years ago
- ☆62Updated last year
- Pytorch Custom CUDA kernel for searchsorted☆136Updated 10 months ago
- Solve sparse linear systems in JAX using the KLU algorithm☆31Updated last week
- Riemannian Convex Potential Maps☆68Updated last year
- numerical optimization in pytorch☆129Updated 4 months ago
- Path Tracing in JAX☆69Updated 4 years ago
- ☆21Updated last year
- ☆22Updated 4 years ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆48Updated 3 weeks ago
- Implicit^2: Implicit model for implicit neural representations☆27Updated 2 years ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆53Updated 2 weeks ago
- PyTorch implementation of paper "Flat Metric Minimization with Applications in Generative Modeling"☆19Updated 5 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆66Updated 3 months ago
- JAX exponential map normalising flows on sphere☆17Updated 3 years ago
- Pytorch implementation of Gauge Equivariant Mesh CNN papers☆9Updated 4 years ago
- Differentiable interface to FEniCS for JAX☆49Updated 3 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.☆25Updated 3 years ago
- Geometry processing utilities compatible with jax for autodifferentiation.☆83Updated 11 months ago