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
  ☆53Updated 11 months ago
• anabatsh / PROTES
  Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) f…
  ☆33Updated last month
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆59Updated last week
• savostyanov / ttcross
  Cross interpolation of high-dimensional arrays in tensor train format
  ☆24Updated last year
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆96Updated last year
• eepperly / XTrace
  Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"
  ☆14Updated 2 years ago
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆66Updated this week
• RehMoritz / vmc_pde
  Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…
  ☆10Updated 2 years ago
• PGelss / scikit_tt
  Tensor Train Toolbox
  ☆114Updated this week
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆73Updated 2 months ago
• AndreiChertkov / ttopt
  Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…
  ☆36Updated last year
• BallisticLA / parla
  Python Algorithms for Randomized Linear Algebra
  ☆54Updated 2 years ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• scalable-matrix / H2Pack
  H2 Matrix Package
  ☆30Updated last year
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆97Updated last week
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆125Updated last month
• qiyang-ustc / jaxeigs
  Rescure useful code in TensorNetwork by google for further personal usage.
  ☆12Updated 6 months ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆111Updated 3 months ago
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆29Updated last month
• rfeinman / Torch-ARPACK
  Fast extremal eigensolvers for PyTorch.
  ☆17Updated 4 years ago
• lanaperisa / TensorToolbox.jl
  Julia package for tensors as multidimensional arrays, with functionalty within Tucker format, Kruskal (CP) format, Hierarchical Tucker fo…
  ☆66Updated 2 years ago
• BallisticLA / RandBLAS
  A header-only C++ library for sketching in randomized linear algebra
  ☆91Updated 3 months ago
• EnzymeAD / Enzyme-JAX
  Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.
  ☆86Updated this week
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆196Updated this week
• danielward27 / paramax
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆34Updated last month
• lukasheinrich / pyhep2020-autodiff-tutorial
  ☆31Updated 4 years ago
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆30Updated last year
• oseledets / ttpy
  Python implementation of the TT-Toolbox
  ☆248Updated 3 months ago
• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆40Updated 7 months ago