ion-g-ion / torchTTLinks
Tensor-Train decomposition in pytorch
☆68Updated 5 months ago
Alternatives and similar repositories for torchTT
Users that are interested in torchTT are comparing it to the libraries listed below
Sorting:
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆72Updated last year
- A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays☆53Updated last year
- Tensor Train Toolbox☆114Updated 2 weeks ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆110Updated 7 months ago
- Tensor Network Learning with PyTorch☆298Updated last year
- A thoroughly investigated survey for tensorial neural networks.☆136Updated 6 months ago
- Python Algorithms for Randomized Linear Algebra☆54Updated 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
- Efficient and Accurate Neural-Network Ansatz for Quantum Monte Carlo☆64Updated 7 months ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 4 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 4 months ago
- JAX accelerated Quantum Monte Carlo☆81Updated last month
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆79Updated last year
- Differentiable scientific computing library☆152Updated 11 months ago
- Automatic High-Order Optimization for Tensors☆23Updated 2 years ago
- High-performance quantum systems simulation with JAX (GPU-accelerated & differentiable solvers).☆231Updated this week
- Riemannian Optimization Using JAX☆49Updated last year
- Smooth integration of tensor networks in machine learning☆44Updated last week
- Turning SymPy expressions into PyTorch modules.☆148Updated last year
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆280Updated 2 weeks ago
- Matrix-free linear algebra in JAX.☆126Updated last month
- Cross factorization of tensors☆37Updated 2 weeks ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆206Updated 7 months ago
- A header-only C++ library for sketching in randomized linear algebra☆91Updated 3 months ago
- Pytorch implementation of KFAC - this is a port of https://github.com/tensorflow/kfac/☆25Updated last year
- GPU-accelerated Monte Carlo simulations of 2D Ising Model☆66Updated 3 months ago
- This is project page for the paper "RG-Flow: a hierarchical and explainable flow model based on renormalization group and sparse prior". …☆86Updated 2 years ago
- Impementation of Variational Monte Carlo (VMC) for quantum many-body dynamics using JAX.☆65Updated 4 months ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- Kolmogorov-Arnold Networks built on JAX☆82Updated 3 months ago