Alternatives and similar repositories for gradML

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

• jvanname / spectralradiinai
  We are applying the notion of the spectral radius to NLP and data represented as graphs.
  ☆10Updated 5 months ago
• patrick-kidger / typst_pyimage
  Typst extension, adding support for generating figures using inline Python code
  ☆109Updated 2 months ago
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆149Updated 2 weeks ago
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆165Updated this week
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆110Updated last year
• normal-computing / thermox
  Exact OU processes with JAX
  ☆57Updated 8 months ago
• awni / picochat
  Smaller and faster nanochat in MLX
  ☆32Updated 2 weeks ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆114Updated 11 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• NERSC / slurm-ray-cluster
  ☆61Updated 2 years ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆44Updated last year
• ekzhang / ekzhang.sty
  My personal LaTeX template, with sensible formatting and commands
  ☆16Updated 3 years ago
• saforem2 / ezpz
  Train across all your devices, ezpz 🍋
  ☆24Updated last week
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆137Updated last year
• jax-ml / jax-ai-stack
  ☆246Updated this week
• probml / JSL
  Jax SSM Library
  ☆48Updated 3 years ago
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆69Updated last month
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 4 months ago
• okarthikb / state-space-models
  ☆28Updated last year
• saforem2 / ambivalent
  Minimal, beautiful (+ highly-customizable) styles for Matplotlib.
  ☆19Updated last week
• probml / ssm-book
  Interactive textbook on state-space models
  ☆199Updated last year
• wonmor / ElectronVisualized
  Public Archive: Beautiful and Elegant Quantum Mechanics Visualization.
  ☆74Updated 3 months ago
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆33Updated 3 years ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆354Updated 7 months ago
• saurabhaloneai / Llama-3-From-Scratch-In-Pure-Jax
  This repository contain the simple llama3 implementation in pure jax.
  ☆70Updated 9 months ago
• ESHyperscale / nano-egg
  Evolution Pretraining Fully in Int Formats
  ☆38Updated last week
• kabouzeid / turm
  TUI for the Slurm Workload Manager
  ☆230Updated 3 months ago
• qqq-wisc / qwla
  Quantum computing without the linear algebra
  ☆76Updated 5 months ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆364Updated this week