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".
  ☆282Updated 7 months ago
• mitmath / matrixcalc
  MIT IAP short course: Matrix Calculus for Machine Learning and Beyond
  ☆570Updated last week
• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆168Updated 8 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
  ☆137Updated last year
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆458Updated 6 months ago
• srush / Autodiff-Puzzles
  ☆492Updated last year
• Atcold / Energy-Book
  Deep Learning, an Energy Approach
  ☆242Updated 8 months ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆506Updated 6 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 5 months ago
• 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
  ☆843Updated 2 weeks ago
• Ma-Lab-Berkeley / deep-representation-learning-book
  Learning Deep Representations of Data Distributions
  ☆765Updated this week
• abao1999 / panda
  Patched Attention for Nonlinear Dynamics
  ☆167Updated last month
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆221Updated last year
• damek / STAT-4830
  Official Github for Wharton STAT 4830
  ☆55Updated this week
• probml / ssm-book
  Interactive textbook on state-space models
  ☆202Updated 2 years ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,860Updated 7 months ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆222Updated last year
• ludocomito / learning-diffusion
  A practical guide to diffusion models, implemented from scratch.
  ☆244Updated last month
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆379Updated last week
• google-deepmind / nanodo
  ☆289Updated last year
• yuanchenyang / smalldiffusion
  Simple and readable code for training and sampling from diffusion models
  ☆696Updated 7 months ago
• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆469Updated 10 months ago
• google / autobound
  AutoBound automatically computes upper and lower bounds on functions.
  ☆364Updated 3 months ago
• Quentin-Anthony / torch-profiling-tutorial
  ☆541Updated 6 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆243Updated 2 months ago
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆207Updated 4 months ago
• google-deepmind / uncertain_ground_truth
  Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…
  ☆678Updated last year
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆672Updated last week
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆406Updated this week