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
  ☆113Updated 2 months ago
• awni / picochat
  Smaller and faster nanochat in MLX
  ☆36Updated 2 months ago
• catswe / LinearKAN
  LinearKAN: A very fast implementation of Kolmogorov-Arnold Networks
  ☆18Updated 2 months ago
• teorth / optimizationproblems
  A collection of optimization problems in mathematics
  ☆194Updated this week
• dcroote / stanford-thesis-example
  Stanford LaTeX thesis template
  ☆58Updated last month
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆110Updated last year
• geohot / dumbrl
  Can RL solve simple problems?
  ☆54Updated 2 years ago
• wellecks / transformers4math-simons
  Transformers for Mathematics Tutorial | Simons/SLMath Workshop on AI for Mathematics 2025
  ☆43Updated 3 weeks ago
• probml / ssm-book
  Interactive textbook on state-space models
  ☆202Updated 2 years ago
• saforem2 / ambivalent
  Minimal, beautiful (+ highly-customizable) styles for Matplotlib.
  ☆19Updated 2 months ago
• Artur-Galstyan / statedict2pytree
  ☆46Updated last week
• Kripner / nanoproof
  Minimal open-source implementation of AlphaProof and HyperTree Proof Search.
  ☆62Updated last week
• mzguntalan / zephyr
  Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulat…
  ☆36Updated 5 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• RylanSchaeffer / Stanford-AI-Alignment-Double-Descent-Tutorial
  Code for Arxiv Double Descent Demystified: Identifying, Interpreting & Ablating the Sources of a Deep Learning Puzzle
  ☆64Updated 2 years ago
• Tobotis / fourier-guide
  An interactive guide to the Fourier-Transform.
  ☆26Updated last year
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆69Updated 4 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• HumanCompatibleAI / leela-interp
  Code for "Evidence of Learned Look-Ahead in a Chess-Playing Neural Network"
  ☆26Updated last year
• HackerCupAI / starter-kits
  ☆68Updated last year
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆191Updated this week
• leehuwuj / python-uv-monorepo-template
  A clean monorepo template for a Python project using uv
  ☆13Updated 7 months ago
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆164Updated last year
• normal-computing / thermox
  Exact OU processes with JAX
  ☆59Updated last month
• Farama-Foundation / Jumpy
  On-the-fly conversions between Jax and NumPy tensors
  ☆56Updated 2 years ago
• Algomancer / VCReg
  Minimal Implimentation of VCRec (2024) for collapse provention.
  ☆18Updated last year
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 8 months ago
• lucidrains / torch-einops-utils
  Some utility functions to help myself (and perhaps others) go faster with ML/AI work
  ☆45Updated this week
• oripress / AlgoTune
  AlgoTune is a NeurIPS 2025 benchmark made up of 154 math, physics, and computer science problems. The goal is write code that solves each…
  ☆88Updated last week