yhtang / FunFact
Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
☆58Updated 2 years ago
Alternatives and similar repositories for FunFact:
Users that are interested in FunFact are comparing it to the libraries listed below
- Python Algorithms for Randomized Linear Algebra☆52Updated last year
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆93Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆24Updated 3 months ago
- Turning SymPy expressions into JAX functions☆44Updated 3 years ago
- Differentiable interface to FEniCS for JAX☆52Updated 3 years ago
- Tensor Train Toolbox☆108Updated last month
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 5 months ago
- Riemannian Optimization Using JAX☆48Updated last year
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- A software package for flexible HPC GPs☆15Updated 2 weeks ago
- ☆20Updated 3 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆103Updated 3 weeks ago
- Python Tensor Toolbox☆29Updated 2 weeks ago
- Stencil computations in JAX☆70Updated last year
- Stiff Neural Ordinary Differential Equations☆32Updated last year
- Matrix-free linear algebra in JAX.☆114Updated last month
- Abstract your array operations.☆142Updated this week
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 8 months ago
- An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py☆58Updated 3 years ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆76Updated 8 months ago
- Jax-based quantum state vector simulator.☆25Updated this week
- Solving Optimization Problems with JAX, code and PDF☆15Updated 4 years ago
- Turning SymPy expressions into PyTorch modules.☆145Updated last year
- Implementation of Forward Laplacian algorithm in JAX☆59Updated 3 weeks ago
- Learning Green's functions of partial differential equations with deep learning.☆63Updated last year
- Automatic Functional Differentiation in JAX☆63Updated last month
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆59Updated 3 months 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
- Multiple dispatch over abstract array types in JAX.☆114Updated 2 weeks ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆59Updated 3 months ago