Linux-cpp-lisp / opt_einsum_fx

Related projects ⓘ

Alternatives and complementary repositories for opt_einsum_fx

• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 3 years ago
• bdusell / semiring-einsum
  Generic PyTorch implementation of einsum that supports different semirings
  ☆46Updated 3 months ago
• JuliaGPU / GemmKernels.jl
  Flexible and performant GEMM kernels in Julia
  ☆79Updated last month
• google-deepmind / einshape
  ☆98Updated 4 months ago
• microsoft / knossos-ksc
  Compiler with automatic differentiation
  ☆45Updated last year
• LinjianMa / AutoHOOT
  Automatic High-Order Optimization for Tensors
  ☆23Updated last year
• EnzymeAD / Enzyme-JAX
  ☆48Updated this week
• duvenaud / jaxde
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Updated 4 years ago
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆25Updated last month
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆40Updated 4 years ago
• NERSC / slurm-ray-cluster
  ☆52Updated last year
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 2 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆43Updated 2 weeks ago
• vardigroup / TensorOrder
  A tool for weighted model counting through tensor network contraction
  ☆20Updated 2 years ago
• JuliaLabs / brutus
  ☆60Updated 9 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆58Updated last year
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• pyro-ppl / pyro-api
  Generic API for dispatch to Pyro backends.
  ☆16Updated 2 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• jaanli / hierarchical-variational-models-physics
  Hierarchical variational models for physics.
  ☆18Updated 4 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆46Updated 3 years ago
• deepchem / jaxchem
  JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling
  ☆79Updated 4 years ago
• riswords / quiche
  ☆22Updated last year
• Zymrael / torchSODE
  PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization
  ☆16Updated 4 years ago
• FluxML / XLA.jl
  "Maybe we have our own magic."
  ☆47Updated 4 years ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆55Updated 2 weeks ago
• Joshuaalbert / jaxnlds
  Inference on non-linear dynamical systems written in JAX
  ☆12Updated 4 years ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• rsokl / phantom-tensors
  Tensor-like types – with variadic shapes – that support both static and runtime type checking, and convenient parsing
  ☆17Updated last week