rsokl / phantom-tensors

Related projects ⓘ

Alternatives and complementary repositories for phantom-tensors

• NeilGirdhar / tjax
  Tools for JAX
  ☆43Updated 2 weeks ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆41Updated 3 weeks ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆55Updated 2 weeks ago
• cgarciae / simple-pytree
  A dead simple Python package for creating custom JAX pytree objects
  ☆61Updated 4 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆66Updated last year
• PhilipVinc / numba4jax
  Use numba-compiled kernels from within Jax
  ☆25Updated last month
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆58Updated last year
• jax-ml / jax-ai-stack
  ☆65Updated this week
• jax-ml / coix
  Inference Combinators in JAX
  ☆43Updated 2 weeks ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆102Updated 3 weeks ago
• google-deepmind / einshape
  ☆98Updated 4 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• pyro-ppl / pyro-api
  Generic API for dispatch to Pyro backends.
  ☆16Updated 2 years ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• srush / learns-dex
  ☆33Updated 3 years ago
• dfm / jpu
  JAX + Units
  ☆43Updated this week
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆33Updated last week
• wesselb / varz
  Painless optimisation of constrained variables in AutoGrad, TensorFlow, PyTorch, and JAX
  ☆23Updated last year
• aesara-devs / aemcmc
  AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.
  ☆39Updated last year
• wesselb / lab
  A generic interface for linear algebra backends
  ☆70Updated 4 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆85Updated 8 months ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆39Updated last year
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆61Updated 4 months ago
• gehring / fax
  ☆78Updated 3 years ago
• EnzymeAD / Enzyme-JAX
  ☆48Updated this week
• JiaYaobo / fenbux
  A Simple Statistical Distribution Library in JAX
  ☆16Updated 7 months ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆80Updated this week
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆84Updated 6 months ago