fasghq / ttaxLinks
Tensor-Train toolbox in Jax
☆11Updated 3 years ago
Alternatives and similar repositories for ttax
Users that are interested in ttax are comparing it to the libraries listed below
Sorting:
- A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays☆52Updated 11 months ago
- Python Algorithms for Randomized Linear Algebra☆54Updated 2 years ago
- Tensor Train Toolbox☆113Updated last month
- Orthogonal polynomials with JAX☆22Updated this week
- H2 Matrix Package☆30Updated last year
- Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) f…☆33Updated last month
- Cross interpolation of high-dimensional arrays in tensor train format☆24Updated last year
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆14Updated 2 years ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆84Updated this week
- A software package for flexible HPC GPs☆16Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆95Updated this week
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆96Updated last year
- Wigner's 3J, 6J, 9J symbols for python☆21Updated 10 months ago
- Julia package for tensors as multidimensional arrays, with functionalty within Tucker format, Kruskal (CP) format, Hierarchical Tucker fo…☆66Updated 2 years ago
- Time Integration Package☆11Updated 5 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆39Updated 7 months ago
- Toolbox for HODLR and HSS matrices in MATLAB☆46Updated 2 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 4 years ago
- Numerical quadrature with JAX☆61Updated this week
- Fast orthogonal polynomial transforms☆62Updated 8 months ago
- Analysis of initial value ODE solvers☆82Updated 9 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
- Implementation of Forward Laplacian algorithm in JAX☆76Updated last week
- JAX bindings for the NVIDIA cuDecomp library☆36Updated last month
- H2Lib public repository☆58Updated 2 years ago
- Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…☆36Updated last year
- Reverse-mode AD of dominant sparse eigensolver using Pytorch.☆41Updated 4 years ago
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated 8 months ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 3 months ago