anabatsh / PROTESLinks
Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) format.
☆38Updated 7 months ago
Alternatives and similar repositories for PROTES
Users that are interested in PROTES 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☆58Updated 3 weeks ago
- Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…☆37Updated 2 years ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- ☆11Updated 5 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Models and code for the ICLR 2020 workshop paper "Towards Understanding Normalization in Neural ODEs"☆16Updated 5 years ago
- Continuous-time gradient flow for generative modeling and variational inference☆33Updated 7 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆39Updated 3 years ago
- Tensor-Train decomposition in pytorch☆81Updated 3 weeks ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 months ago
- Library for normalizing flows and neural flows.☆25Updated 3 years ago
- Normalizing Flows using JAX☆85Updated 2 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
- Differentiable and numerically stable implementation of the matrix exponential☆33Updated 5 years ago
- ☆22Updated last year
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- A differentiation API for PyTorch☆30Updated 5 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆117Updated last month
- Implementation of Forward Laplacian algorithm in JAX☆92Updated 2 weeks ago
- Pytorch implement of the paper Neural Canonical Transformation with Symplectic Flows☆31Updated 5 years ago
- Pytorch source code for arXiv paper Neural Network Renormalization Group, a generative model using variational renormalization group and …☆80Updated 6 years ago
- ☆52Updated 2 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 3 years ago
- Cross interpolation of high-dimensional arrays in tensor train format☆26Updated last year
- Combining smooth constraint for building DAG with normalizing flow in order to replace autoregressive transformations while keeping tract…☆46Updated 2 years ago