Alternatives and similar repositories for tensorgrad:

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

• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆406Updated last month
• srush / Autodiff-Puzzles
  ☆428Updated 6 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆60Updated 8 months ago
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆237Updated 2 weeks ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆222Updated 9 months ago
• che-shr-cat / JAX-in-Action
  Notebooks for the "JAX in Action" book
  ☆142Updated 9 months ago
• mlecauchois / micrograd-cuda
  ☆242Updated last year
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆199Updated last month
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆270Updated 10 months ago
• misko / human_descent
  ☆36Updated 4 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆188Updated 3 weeks ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆472Updated 2 weeks ago
• francoisfleuret / dlc
  ☆41Updated 3 months ago
• AI-Hypercomputer / RecML
  ☆86Updated this week
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆342Updated last year
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆375Updated this week
• google-deepmind / nanodo
  ☆215Updated 9 months ago
• BlackHC / neural_net_checklist
  ☆150Updated 8 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆507Updated this week
• policy-gradient / GRPO-Zero
  ☆238Updated this week
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆544Updated this week
• yuanchenyang / smalldiffusion
  Simple and readable code for training and sampling from diffusion models
  ☆471Updated 3 months ago
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆198Updated last year
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆187Updated 11 months ago
• unixpickle / learn-ptx
  Learning about CUDA by writing PTX code.
  ☆127Updated last year
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆365Updated last week
• bhosmer / mm
  ☆130Updated 5 months ago
• HMUNACHI / NanoDL
  A Jax-based library for designing and training small transformers.
  ☆286Updated 7 months ago
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆348Updated this week
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆332Updated last month