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".
  ☆280Updated 6 months ago
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆457Updated 4 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• fbach2000 / Learning_Theory_from_First_Principles
  Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach
  ☆133Updated last year
• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆165Updated 6 months ago
• mitmath / matrixcalc
  MIT IAP short course: Matrix Calculus for Machine Learning and Beyond
  ☆561Updated last week
• Atcold / Energy-Book
  Deep Learning, an Energy Approach
  ☆226Updated 6 months ago
• Ma-Lab-Berkeley / deep-representation-learning-book
  Learning Deep Representations of Data Distributions
  ☆725Updated this week
• srush / Autodiff-Puzzles
  ☆460Updated last year
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆781Updated this week
• wilson-labs / cola
  Compositional Linear Algebra
  ☆504Updated 4 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆316Updated 4 months ago
• tugot17 / pmpp
  Complete solutions to the Programming Massively Parallel Processors Edition 4
  ☆611Updated 6 months ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆222Updated last year
• Quentin-Anthony / torch-profiling-tutorial
  ☆533Updated 4 months ago
• ludocomito / learning-diffusion
  A practical guide to diffusion models, implemented from scratch.
  ☆223Updated 2 weeks ago
• francoisfleuret / dlc
  ☆42Updated last year
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,831Updated 6 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆235Updated last month
• probml / ssm-book
  Interactive textbook on state-space models
  ☆200Updated last year
• google-deepmind / nanodo
  ☆286Updated last year
• MilesCranmer / awesome-ml-demos
  Curated list of interactive ML demos
  ☆356Updated 9 months ago
• mlecauchois / micrograd-cuda
  ☆249Updated last year
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆206Updated 2 months ago
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆221Updated last year
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆662Updated this week
• google-deepmind / torax
  TORAX: Tokamak transport simulation in JAX
  ☆581Updated last week
• abao1999 / panda
  Patched Attention for Nonlinear Dynamics
  ☆162Updated last week
• google-deepmind / uncertain_ground_truth
  Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…
  ☆679Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆299Updated last year