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 this week
- A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays☆50Updated 8 months ago
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆58Updated 2 years ago
- A thoroughly investigated survey for tensorial neural networks.☆125Updated 2 months ago
- Code for "Symplectic Adjoint Method for Exact Gradient of Neural ODE with Minimal Memory," NeurIPS, 2021.☆16Updated 3 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…☆34Updated last year
- ☆20Updated 4 months ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆102Updated 3 months ago
- ☆60Updated 4 years ago
- A JAX based package designed for efficient second order operators (e.g., laplacian) computation.☆71Updated last year
- Inverse Rosenblatt Transform (Conditional Distribution) + MCMC sampling using Tensor Train approximation☆14Updated 3 weeks ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆77Updated 9 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- JAX accelerated Quantum Monte Carlo☆72Updated last week
- Riemannian Optimization Using JAX☆48Updated last year
- Python Algorithms for Randomized Linear Algebra☆53Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆105Updated this week
- Efficient and Accurate Neural-Network Ansatz for Quantum Monte Carlo☆58Updated 3 months ago
- Tensor Network Learning with PyTorch☆10Updated 5 years ago
- Tensor Network Learning with PyTorch☆293Updated 9 months ago
- Matlab Algorithms for Randomized Linear Algebra☆15Updated last year
- Automatic Functional Differentiation in JAX☆64Updated 2 months ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 3 years ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆59Updated 4 months ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 weeks ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆70Updated 9 months ago
- Turning SymPy expressions into PyTorch modules.☆146Updated last year
- ☆246Updated 6 months ago
- Pytorch source code for arXiv paper Neural Network Renormalization Group, a generative model using variational renormalization group and …☆79Updated 5 years ago