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
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆93Updated last year
- Python Algorithms for Randomized Linear Algebra☆53Updated last year
- Python Tensor Toolbox☆30Updated this week
- An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py☆58Updated 3 years ago
- Differentiable interface to FEniCS for JAX☆53Updated 3 years ago
- Turning SymPy expressions into JAX functions☆44Updated 4 years ago
- Tensor Train Toolbox☆108Updated last week
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆77Updated 9 months ago
- Solving Optimization Problems with JAX, code and PDF☆15Updated 4 years ago
- Learning Green's functions of partial differential equations with deep learning.☆65Updated last year
- A software package for flexible HPC GPs☆16Updated 3 weeks ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 3 years ago
- H2 Matrix Package☆29Updated last year
- Stiff Neural Ordinary Differential Equations☆32Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆123Updated 6 months ago
- Stencil computations in JAX☆70Updated last year
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 9 months ago
- PyTorch implementation of GMLS-Nets. Machine learning methods for scattered unstructured data sets. Methods for learning differential op…☆26Updated last year
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆59Updated 4 months ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆41Updated 6 months ago
- Datasets and code for results presented in the ProbConserv paper☆53Updated 9 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆27Updated 4 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
- Convolutional Solvers for Partial Differential Equations☆28Updated 4 years ago
- Turning SymPy expressions into PyTorch modules.☆146Updated last year
- A Python library for training neural ODEs.☆20Updated last month
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- A short course in pseudospectral collocation methods for wave equations, with implementations in Python.☆51Updated 2 years ago
- Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed c…☆114Updated 3 years ago