Alternatives and similar repositories for tensorgrad

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

• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆150Updated last month
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆425Updated 2 months ago
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆424Updated this week
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆241Updated 3 weeks ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated 11 months ago
• srush / Autodiff-Puzzles
  ☆440Updated 8 months ago
• Atcold / Energy-Book
  Deep Learning, an Energy Approach
  ☆187Updated last month
• mitmath / matrixcalc
  MIT IAP short course: Matrix Calculus for Machine Learning and Beyond
  ☆512Updated 5 months ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆224Updated last year
• wilson-labs / cola
  Compositional Linear Algebra
  ☆478Updated last month
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆203Updated 4 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆205Updated 2 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆204Updated 3 months ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,803Updated 3 weeks ago
• mlecauchois / micrograd-cuda
  ☆248Updated last year
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆555Updated last week
• fbach2000 / Learning_Theory_from_First_Principles
  Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach
  ☆114Updated last year
• google-deepmind / nanodo
  ☆273Updated last year
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆210Updated last year
• joelburget / microjax
  A tiny autograd engine with a Jax-like API
  ☆68Updated last week
• tugot17 / pmpp
  Complete solutions to the Programming Massively Parallel Processors Edition 4
  ☆196Updated 3 weeks ago
• yuanchenyang / smalldiffusion
  Simple and readable code for training and sampling from diffusion models
  ☆511Updated last month
• google-deepmind / torax
  TORAX: Tokamak transport simulation in JAX
  ☆436Updated last week
• xbresson / CS5242_2025
  NUS CS5242 Neural Networks and Deep Learning, Xavier Bresson, 2025
  ☆393Updated 2 months ago
• misko / human_descent
  ☆36Updated 7 months ago
• abao1999 / panda
  Patched Attention for Nonlinear Dynamics
  ☆150Updated 2 weeks ago
• obadakhalili / tinygrad-tensor-puzzles
  Solve puzzles to improve your tinygrad skills!
  ☆135Updated 4 months ago
• Foundations-of-Computer-Vision / visionbook
  <Foundations of Computer Vision> Book
  ☆265Updated 3 weeks ago
• google-deepmind / uncertain_ground_truth
  Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…
  ☆650Updated last year
• opallab / neural_networks_and_computation
  Computational abilities and efficiency of neural networks
  ☆53Updated 3 weeks ago