rfeinman / Torch-ARPACK
Fast extremal eigensolvers for PyTorch.
☆17Updated 3 years ago
Alternatives and similar repositories for Torch-ARPACK:
Users that are interested in Torch-ARPACK are comparing it to the libraries listed below
- Differentiable interface to FEniCS for JAX☆52Updated 3 years ago
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆12Updated last year
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆92Updated last year
- Reverse-mode AD of dominant sparse eigensolver using Pytorch.☆38Updated 4 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆53Updated last month
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆9Updated 2 years ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆58Updated 2 months ago
- Python Algorithms for Randomized Linear Algebra☆50Updated last year
- A short course in pseudospectral collocation methods for wave equations, with implementations in Python.☆50Updated last year
- Analysis of initial value ODE solvers☆80Updated 4 months ago
- Easy interoperability with Automatic Differentiation libraries through NumPy interface to Firedrake and FEniCS☆14Updated last year
- Learning Green's functions of partial differential equations with deep learning.☆63Updated last year
- Code for the paper "Rational neural networks", NeurIPS 2020☆28Updated 3 years ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 2 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆22Updated 2 months ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 months ago
- Code to estimate Renormalized Mutual Information in simple settings☆12Updated 3 years ago
- Differentiable interface to Firedrake for JAX☆14Updated 3 years ago
- H2 Matrix Package☆25Updated last year
- Stiff Neural Ordinary Differential Equations☆32Updated last year
- A software package for flexible HPC GPs☆14Updated last month
- Orthogonal polynomials with JAX☆12Updated 2 weeks ago
- Solve sparse linear systems in JAX using the KLU algorithm☆34Updated 2 months ago
- Toolbox for HODLR and HSS matrices in MATLAB☆45Updated last year
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆67Updated 5 months ago
- ☆31Updated 4 years ago
- ☆11Updated 4 months ago
- ☆12Updated 6 years ago
- Matrix-free linear algebra in JAX.☆113Updated last week
- Fast orthogonal polynomial transforms☆58Updated 4 months ago