fasghq / ttax
Tensor-Train toolbox in Jax
☆10Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for ttax
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆89Updated last year
- A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays☆44Updated 4 months ago
- Tensor Train Toolbox☆104Updated this week
- Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…☆30Updated last year
- Cross interpolation of high-dimensional arrays in tensor train format☆21Updated 9 months ago
- Analysis of initial value ODE solvers☆78Updated 2 months ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆47Updated 8 months ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 3 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated this week
- H2 Matrix Package☆24Updated last year
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Numerical quadrature with JAX☆43Updated 2 weeks ago
- Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) f…☆30Updated 2 weeks ago
- Python implementation of the TT-Toolbox☆240Updated 9 months ago
- Automatic High-Order Optimization for Tensors☆23Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆93Updated 2 months ago
- Stencil computations in JAX☆66Updated last year
- Numerical integration of Ito or Stratonovich SDEs☆157Updated last year
- Differentiable interface to FEniCS for JAX☆50Updated 3 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- Solve sparse linear systems in JAX using the KLU algorithm☆32Updated this week
- Models and code for the ICLR 2020 workshop paper "Towards Understanding Normalization in Neural ODEs"☆16Updated 4 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆34Updated 2 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆120Updated last month
- Toolbox for HODLR and HSS matrices in MATLAB☆43Updated last year
- Python Algorithms for Randomized Linear Algebra☆46Updated last year
- A Python package for barycentric rational and polynomial interpolation and approximation.☆44Updated 5 months ago
- ☆11Updated 3 years ago
- Matrix-free linear algebra in JAX.☆103Updated 2 months ago
- A short course in pseudospectral collocation methods for wave equations, with implementations in Python.☆48Updated last year