Alternatives and similar repositories for ttax:

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

• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆93Updated last year
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆53Updated 3 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 4 years ago
• scalable-matrix / H2Pack
  H2 Matrix Package
  ☆29Updated last year
• AndreiChertkov / teneva
  A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays
  ☆52Updated 9 months ago
• savostyanov / ttcross
  Cross interpolation of high-dimensional arrays in tensor train format
  ☆24Updated last year
• Pranab-JD / LeXInt
  Time Integration Package
  ☆11Updated 4 months ago
• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆28Updated 5 months ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆55Updated 4 months ago
• PGelss / scikit_tt
  Tensor Train Toolbox
  ☆110Updated last month
• 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
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆16Updated last month
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆26Updated 2 months ago
• anabatsh / PROTES
  Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) f…
  ☆32Updated 3 months ago
• H2Lib / H2Lib
  H2Lib public repository
  ☆56Updated 2 years ago
• EnzymeAD / Enzyme-JAX
  Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.
  ☆75Updated this week
• jonas-schmitt / evostencils
  Automating the Design of Multigrid Methods with Evolutionary Program Synthesis
  ☆13Updated 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…
  ☆35Updated last year
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆71Updated 3 weeks ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆116Updated 3 months ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆90Updated last week
• numpi / hm-toolbox
  Toolbox for HODLR and HSS matrices in MATLAB
  ☆46Updated last week
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆17Updated last week
• ketch / nodepy
  Analysis of initial value ODE solvers
  ☆81Updated 7 months ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆107Updated last month
• BallisticLA / parla
  Python Algorithms for Randomized Linear Algebra
  ☆53Updated last year
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆56Updated last week
• lanaperisa / TensorToolbox.jl
  Julia package for tensors as multidimensional arrays, with functionalty within Tucker format, Kruskal (CP) format, Hierarchical Tucker fo…
  ☆65Updated last year
• LinjianMa / AutoHOOT
  Automatic High-Order Optimization for Tensors
  ☆23Updated 2 years ago