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
  ☆107Updated last month
• saforem2 / ambivalent
  Minimal, beautiful (+ highly-customizable) styles for Matplotlib.
  ☆19Updated 3 months ago
• awni / picochat
  Smaller and faster nanochat in MLX
  ☆28Updated 3 weeks ago
• alugowski / matrepr
  Format matrices and tensors to HTML, string, and LaTeX, with Jupyter integration.
  ☆14Updated 11 months ago
• sifakis / CS639S23_Demos
  Software artifacts and Demos for CS639 (Spring 2023) "Parallel and Throughput-Optimized Programming"
  ☆18Updated 2 years ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆43Updated last year
• profConradi / ALife
  Artificial Life simulations
  ☆54Updated 8 months ago
• wonmor / ElectronVisualized
  Public Archive: Beautiful and Elegant Quantum Mechanics Visualization.
  ☆74Updated 3 months ago
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆68Updated last month
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆148Updated 6 months ago
• bhosmer / mm
  ☆138Updated last year
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 4 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆109Updated last year
• probml / JSL
  Jax SSM Library
  ☆48Updated 2 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆113Updated 10 months ago
• chrisby / prometheusCV
  The non-template for academic CVs in latex
  ☆58Updated 5 years ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆34Updated last year
• wellecks / transformers4math-simons
  Transformers for Mathematics Tutorial | Simons/SLMath Workshop on AI for Mathematics 2025
  ☆41Updated 7 months ago
• normal-computing / thermox
  Exact OU processes with JAX
  ☆56Updated 7 months ago
• rapidsai / cugraph-gnn
  ☆56Updated last week
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• ekzhang / ekzhang.sty
  My personal LaTeX template, with sensible formatting and commands
  ☆16Updated 3 years ago
• Benny-Nottonson / Mojo-Marathons
  ☆28Updated last year
• TysonRayJones / PythonTools
  Tools I've written in Python which are useful/necessary for my research
  ☆10Updated 7 years ago
• arXiv / arxiv-view-as-html
  Repository for the view as html feature
  ☆28Updated last week
• yobibyte / compressor
  Because we don't have enough time to read everything
  ☆89Updated last year
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆71Updated 5 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆136Updated 3 weeks ago