anabatsh / PROTESLinks
Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) format.
☆37Updated 4 months ago
Alternatives and similar repositories for PROTES
Users that are interested in PROTES are comparing it to the libraries listed below
Sorting:
- Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…☆36Updated 2 years ago
- A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays☆57Updated last year
- ☆11Updated 4 years ago
- Riemannian Optimization Using JAX☆52Updated last year
- Models and code for the ICLR 2020 workshop paper "Towards Understanding Normalization in Neural ODEs"☆16Updated 5 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆35Updated 3 years ago
- Cross interpolation of high-dimensional arrays in tensor train format☆25Updated last year
- Tensor-Train decomposition in pytorch☆73Updated 7 months ago
- Continuous-time gradient flow for generative modeling and variational inference☆33Updated 6 years ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 3 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆114Updated 6 months ago
- Tensor Train Toolbox☆120Updated last week
- ☆110Updated 4 years ago
- Normalizing Flows using JAX☆84Updated last year
- Differentiable scientific computing library☆154Updated last year
- Matrix-free linear algebra in JAX.☆140Updated 3 weeks ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆101Updated last year
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- orbital MCMC☆10Updated 4 years ago
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆10Updated 3 years ago
- ☆51Updated 2 years ago
- Implementation of Action Matching☆45Updated 2 years ago
- Tensor Network Learning with PyTorch☆299Updated last year
- ☆22Updated 11 months ago
- Convex potential flows☆84Updated 3 years ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆171Updated 3 years ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 6 months ago