Alternatives and similar repositories for tensorgrad:

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

• srush / Autodiff-Puzzles
  ☆416Updated 4 months ago
• che-shr-cat / JAX-in-Action
  Notebooks for the "JAX in Action" book
  ☆130Updated 7 months ago
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆391Updated this week
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆59Updated 6 months ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆219Updated 7 months ago
• google-deepmind / uncertain_ground_truth
  Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…
  ☆644Updated 10 months ago
• google-deepmind / nanodo
  ☆211Updated 7 months ago
• obadakhalili / tinygrad-tensor-puzzles
  Solve puzzles to improve your tinygrad skills!
  ☆111Updated 5 months ago
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆332Updated last year
• salykova / matmul.c
  High-Performance FP32 Matrix Multiplication on CPU
  ☆333Updated this week
• wilson-labs / cola
  Compositional Linear Algebra
  ☆462Updated 2 weeks ago
• misko / human_descent
  ☆36Updated 2 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆265Updated 8 months ago
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆194Updated 9 months ago
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆523Updated this week
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆351Updated 3 weeks ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆136Updated 7 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆491Updated this week
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆269Updated this week
• mlecauchois / micrograd-cuda
  ☆239Updated 11 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆354Updated last week
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆281Updated 5 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆62Updated last year
• francoisfleuret / dlc
  ☆41Updated last month
• modula-systems / modula
  🧱 Modula software package
  ☆145Updated this week
• ulrichstern / cuda-convnet
  Alex Krizhevsky's original code from Google Code
  ☆189Updated 8 years ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆161Updated this week
• revalo / tree-diffusion
  Diffusion on syntax trees for program synthesis
  ☆442Updated 7 months ago