Alternatives and similar repositories for ttax:

Users that are interested in ttax are comparing it to the libraries listed below

• AndreiChertkov / teneva
  A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays
  ☆50Updated 7 months ago
• AndreiChertkov / ttopt
  Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…
  ☆33Updated last year
• anabatsh / PROTES
  Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) f…
  ☆32Updated last month
• savostyanov / ttcross
  Cross interpolation of high-dimensional arrays in tensor train format
  ☆22Updated last year
• PGelss / scikit_tt
  Tensor Train Toolbox
  ☆108Updated last month
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆93Updated last year
• scalable-matrix / H2Pack
  H2 Matrix Package
  ☆27Updated last year
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 3 years ago
• RehMoritz / vmc_pde
  Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…
  ☆9Updated 2 years ago
• BallisticLA / RandBLAS
  A header-only C++ library for sketching in randomized linear algebra
  ☆83Updated this week
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆25Updated 3 weeks ago
• H2Lib / H2Lib
  H2Lib public repository
  ☆53Updated 2 years ago
• ion-g-ion / torchTT
  Tensor-Train decomposition in pytorch
  ☆58Updated 3 weeks ago
• OPTAMI / OPTAMI
  This package is dedicated to high-order optimization methods. All the methods can be used similarly to standard PyTorch optimizers.
  ☆22Updated last month
• EnzymeAD / Enzyme-JAX
  Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.
  ☆60Updated this week
• ketch / nodepy
  Analysis of initial value ODE solvers
  ☆80Updated 5 months ago
• BallisticLA / parla
  Python Algorithms for Randomized Linear Algebra
  ☆52Updated last year
• necludov / wqmc
  ☆14Updated last year
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆89Updated 2 weeks ago
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆15Updated 3 weeks ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆55Updated 2 months ago
• LinjianMa / AutoHOOT
  Automatic High-Order Optimization for Tensors
  ☆23Updated last year
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆63Updated last month
• lanaperisa / TensorToolbox.jl
  Julia package for tensors as multidimensional arrays, with functionalty within Tucker format, Kruskal (CP) format, Hierarchical Tucker fo…
  ☆65Updated last year
• yhtang / FunFact
  Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
  ☆58Updated 2 years ago
• springer13 / tcl
  Tensor Contraction C++ Library
  ☆51Updated 5 years ago
• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆24Updated 3 months ago
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆52Updated 3 years ago
• y1xiaoc / fwdlap
  Forward mode laplacian implemented in JAX tracer
  ☆29Updated 3 months ago