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
- Solving Optimization Problems with JAX, code and PDF☆15Updated 4 years ago
- Differentiable interface to FEniCS for JAX☆50Updated 3 years ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆89Updated 11 months ago
- Tensor Train Toolbox☆104Updated this week
- Stiff Neural Ordinary Differential Equations☆30Updated last year
- Fast extremal eigensolvers for PyTorch.☆17Updated 3 years ago
- Riemannian Optimization Using JAX☆45Updated last year
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆75Updated 5 months ago
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆12Updated last year
- Code to estimate Renormalized Mutual Information in simple settings☆12Updated 3 years ago
- Matlab Algorithms for Randomized Linear Algebra☆14Updated last year
- A header-only C++ library for sketching in randomized linear algebra☆79Updated 2 weeks 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
- Matrix-free linear algebra in JAX.☆101Updated 2 months ago
- Automatic High-Order Optimization for Tensors☆23Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆120Updated last month
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆93Updated 2 months ago
- ☆101Updated 3 years ago
- A Python package for efficient optimisation of real-space renormalization group transformations using Tensorflow.☆27Updated 3 weeks ago
- An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py☆57Updated 3 years ago
- Stencil computations in JAX☆66Updated last year
- Material for 'Mathematics of Deep Learning Workshop' (Invited Talk)☆17Updated last year
- Turning SymPy expressions into PyTorch modules.☆141Updated last year
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 2 years ago
- PyTorch implementation of GMLS-Nets. Machine learning methods for scattered unstructured data sets. Methods for learning differential op…☆23Updated last year
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated last year
- Analysis of initial value ODE solvers☆78Updated 2 months ago
- SciML Benchmarking Suite for AI for Science☆38Updated 3 months ago