yhtang / FunFactLinks
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
Sorting:
- Stencil computations in JAX☆71Updated last year
- Differentiable interface to FEniCS for JAX☆54Updated 4 years ago
- Python Algorithms for Randomized Linear Algebra☆54Updated 2 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆96Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆128Updated 8 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Solving Optimization Problems with JAX, code and PDF☆16Updated 5 years ago
- ☆21Updated 7 months ago
- A software package for flexible HPC GPs☆16Updated last week
- Tensor Train Toolbox☆113Updated last month
- Python Tensor Toolbox☆33Updated last month
- Matrix-free linear algebra in JAX.☆119Updated last week
- Efficient Differentiable n-d PDE solvers in JAX.☆39Updated 7 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 2 months ago
- Riemannian Optimization Using JAX☆49Updated last year
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆10Updated 2 years ago
- Implementation of Forward Laplacian algorithm in JAX☆76Updated last week
- Fast extremal eigensolvers for PyTorch.☆17Updated 4 years ago
- Mathematical operations for JAX pytrees☆198Updated 6 months ago
- Multiple dispatch over abstract array types in JAX.☆124Updated this week
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py☆61Updated 4 years ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 3 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".