Alternatives and similar repositories for tensorgrad

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

• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆273Updated 4 months ago
• fbach2000 / Learning_Theory_from_First_Principles
  Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach
  ☆128Updated last year
• Ma-Lab-Berkeley / deep-representation-learning-book
  Learning Deep Representations of Data Distributions
  ☆601Updated this week
• Atcold / Energy-Book
  Deep Learning, an Energy Approach
  ☆220Updated 5 months ago
• mitmath / matrixcalc
  MIT IAP short course: Matrix Calculus for Machine Learning and Beyond
  ☆547Updated 9 months ago
• srush / Autodiff-Puzzles
  ☆457Updated last year
• wilson-labs / cola
  Compositional Linear Algebra
  ☆491Updated 3 months ago
• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆158Updated 5 months ago
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆451Updated 3 months ago
• tugot17 / pmpp
  Complete solutions to the Programming Massively Parallel Processors Edition 4
  ☆571Updated 4 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆230Updated last week
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆217Updated last year
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆683Updated this week
• modula-systems / modula
  🧱 Modula software package
  ☆303Updated 2 months ago
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆656Updated this week
• google-deepmind / torax
  TORAX: Tokamak transport simulation in JAX
  ☆558Updated last week
• google / autobound
  AutoBound automatically computes upper and lower bounds on functions.
  ☆362Updated 3 weeks ago
• abao1999 / panda
  Patched Attention for Nonlinear Dynamics
  ☆161Updated this week
• misko / human_descent
  ☆37Updated this week
• xbresson / CS5242_2025
  NUS CS5242 Neural Networks and Deep Learning, Xavier Bresson, 2025
  ☆403Updated 6 months ago
• probml / ssm-book
  Interactive textbook on state-space models
  ☆198Updated last year
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆375Updated last month
• Quentin-Anthony / torch-profiling-tutorial
  ☆525Updated 3 months ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆223Updated last year
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆204Updated last month
• yuanchenyang / smalldiffusion
  Simple and readable code for training and sampling from diffusion models
  ☆652Updated 5 months ago
• google-deepmind / nanodo
  ☆285Updated last year
• MilesCranmer / awesome-ml-demos
  Curated list of interactive ML demos
  ☆356Updated 7 months ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,824Updated 4 months ago