BallisticLA / parla
Python Algorithms for Randomized Linear Algebra
☆53Updated 2 years ago
Alternatives and similar repositories for parla:
Users that are interested in parla are comparing it to the libraries listed below
- A header-only C++ library for sketching in randomized linear algebra☆89Updated last month
- Matlab Algorithms for Randomized Linear Algebra☆15Updated last year
- A high-performance C++ library for randomized numerical linear algebra☆68Updated this week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated last month
- Matrix-free linear algebra in JAX.☆118Updated 4 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆28Updated 6 months ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 3 years ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆126Updated 7 months ago
- Differentiable interface to FEniCS for JAX☆53Updated 3 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆94Updated last year
- Stencil computations in JAX☆70Updated last year
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆57Updated last year
- H2 Matrix Package☆30Updated last year
- Python Tensor Toolbox☆33Updated 3 weeks ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆91Updated this week
- A software package for flexible HPC GPs☆16Updated this week
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆25Updated last year
- Code for the book "The Elements of Differentiable Programming".☆84Updated last month
- Multiple dispatch over abstract array types in JAX.☆120Updated last week
- Tensor Train Toolbox☆110Updated last week
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆203Updated 5 months ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆57Updated 5 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆46Updated last month
- Riemannian Optimization Using JAX☆49Updated last year
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆62Updated 6 months ago
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆10Updated 2 years ago
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆14Updated 2 years ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆41Updated last year