Alternatives and similar repositories for gradML

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

• patrick-kidger / typst_pyimage
  Typst extension, adding support for generating figures using inline Python code
  ☆112Updated 2 weeks ago
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 6 months ago
• Artur-Galstyan / statedict2pytree
  ☆44Updated last month
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆109Updated last year
• cgarciae / einop
  ☆60Updated 3 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• jvanname / spectralradiinai
  We are applying the notion of the spectral radius to NLP and data represented as graphs.
  ☆10Updated 6 months ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆45Updated 2 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆114Updated last year
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆70Updated 2 months ago
• ekzhang / ekzhang.sty
  My personal LaTeX template, with sensible formatting and commands
  ☆16Updated 3 years ago
• geohot / tinydreamer
  An implementation of delta-iris in tinygrad
  ☆72Updated last year
• awni / picochat
  Smaller and faster nanochat in MLX
  ☆32Updated last month
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆137Updated this week
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆152Updated last month
• normal-computing / thermox
  Exact OU processes with JAX
  ☆58Updated 9 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 5 months ago
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆171Updated last week
• jax-ml / jax-ai-stack
  ☆255Updated this week
• google / init2winit
  ☆79Updated last week
• cgarciae / jax-gravity
  ☆24Updated 2 years ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆110Updated last month
• geohot / dumbrl
  Can RL solve simple problems?
  ☆54Updated last year
• PrimeIntellect-ai / pi-quant
  SIMD quantization kernels
  ☆93Updated 3 months ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year
• francoisfleuret / picogpt
  Minimal GPT (~350 lines with a simple task to test it)
  ☆63Updated 3 weeks ago
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆48Updated 11 months ago