ion-g-ion / torchTT
Tensor-Train decomposition in pytorch
☆60Updated last month
Alternatives and similar repositories for torchTT:
Users that are interested in torchTT are comparing it to the libraries listed below
- Tensor Train Toolbox☆108Updated last week
- A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays☆50Updated 8 months ago
- A thoroughly investigated survey for tensorial neural networks.☆126Updated 2 months ago
- Code for "Symplectic Adjoint Method for Exact Gradient of Neural ODE with Minimal Memory," NeurIPS, 2021.☆16Updated 3 years ago
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆71Updated last year
- Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…☆35Updated last year
- Tensor Network Learning with PyTorch☆10Updated 6 years ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆77Updated 9 months ago
- Efficient and Accurate Neural-Network Ansatz for Quantum Monte Carlo☆58Updated 3 months ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆102Updated 3 months ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆59Updated 4 months ago
- Tensor Network Learning with PyTorch☆295Updated 10 months ago
- Inverse Rosenblatt Transform (Conditional Distribution) + MCMC sampling using Tensor Train approximation☆14Updated this week
- Python Algorithms for Randomized Linear Algebra☆53Updated last year
- JAX accelerated Quantum Monte Carlo☆72Updated last week
- ☆60Updated 4 years ago
- Cross interpolation of high-dimensional arrays in tensor train format☆24Updated last year
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) f…☆32Updated 2 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆106Updated last week
- Apply CP, Tucker, TT/TR, HT to compress neural networks. Train from scratch.☆12Updated 4 years ago
- ☆20Updated 5 months ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 3 weeks ago
- Python implementation of the TT-Toolbox☆246Updated last week
- TedNet: A Pytorch Toolkit for Tensor Decomposition Networks☆94Updated 2 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Matlab Algorithms for Randomized Linear Algebra☆15Updated last year
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 9 months ago
- PyTorch toolbox for matrix product state models☆144Updated 11 months ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆37Updated last year