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
  ☆48Updated 6 months ago
• anabatsh / PROTES
  Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) f…
  ☆30Updated this week
• AndreiChertkov / ttopt
  Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle…
  ☆32Updated last year
• PGelss / scikit_tt
  Tensor Train Toolbox
  ☆107Updated last month
• savostyanov / ttcross
  Cross interpolation of high-dimensional arrays in tensor train format
  ☆22Updated 11 months ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆92Updated last year
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆12Updated 2 weeks ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 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
• lanaperisa / TensorToolbox.jl
  Julia package for tensors as multidimensional arrays, with functionalty within Tucker format, Kruskal (CP) format, Hierarchical Tucker fo…
  ☆65Updated last year
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆52Updated 3 years ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆84Updated last week
• fujiisoup / py3nj
  Wigner's 3J, 6J, 9J symbols for python
  ☆19Updated 6 months ago
• EnzymeAD / Enzyme-JAX
  ☆53Updated this week
• rfeinman / Torch-ARPACK
  Fast extremal eigensolvers for PyTorch.
  ☆17Updated 3 years ago
• y1xiaoc / fwdlap
  Forward mode laplacian implemented in JAX tracer
  ☆29Updated 2 months ago
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆62Updated 6 months ago
• sandialabs / pyttb
  Python Tensor Toolbox
  ☆28Updated last week
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆50Updated last week
• ion-g-ion / torchTT
  Tensor-Train decomposition in pytorch
  ☆56Updated last month
• solomonik / ctf
  Fork of cyclops-community/ctf repository updated haphazardly, previously this was main repo location
  ☆9Updated 6 years ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆124Updated 4 months ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆53Updated last month
• ketch / PseudoSpectralPython
  A short course in pseudospectral collocation methods for wave equations, with implementations in Python.
  ☆50Updated last year
• lukasheinrich / pyhep2020-autodiff-tutorial
  ☆31Updated 4 years ago
• dolgov / TT-IRT
  Inverse Rosenblatt Transform (Conditional Distribution) + MCMC sampling using Tensor Train approximation
  ☆14Updated last year
• scalable-matrix / H2Pack
  H2 Matrix Package
  ☆25Updated last year
• yhtang / FunFact
  Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
  ☆58Updated 2 years ago
• moble / spherical
  Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics
  ☆46Updated this week