yhtang / FunFact
Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
☆57Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for FunFact
- Python Algorithms for Randomized Linear Algebra☆46Updated last year
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆90Updated last year
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Tensor Train Toolbox☆105Updated last week
- Riemannian Optimization Using JAX☆45Updated last year
- Differentiable interface to FEniCS for JAX☆50Updated 3 years ago
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆121Updated 2 months ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 3 years ago
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆9Updated 2 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆48Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- Turning SymPy expressions into PyTorch modules.☆142Updated last year
- An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py☆57Updated 3 years ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆76Updated 5 months ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆55Updated 2 weeks ago
- Learning Green's functions of partial differential equations with deep learning.☆63Updated 10 months ago
- Solving Optimization Problems with JAX, code and PDF☆15Updated 4 years ago
- Stiff Neural Ordinary Differential Equations☆30Updated last year
- Mathematical operations for JAX pytrees☆189Updated 6 months ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆21Updated 6 months ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 2 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆47Updated 9 months ago
- Stencil computations in JAX☆66Updated last year
- H2 Matrix Package☆24Updated last year
- Reverse-mode AD of dominant sparse eigensolver using Pytorch.☆38Updated 4 years ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆190Updated 4 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated last year
- Implementation of Forward Laplacian algorithm in JAX☆55Updated 2 weeks ago
- Datasets and code for results presented in the ProbConserv paper☆51Updated 5 months ago