Alternatives and similar repositories for jax-from-scratch

Users that are interested in jax-from-scratch are comparing it to the libraries listed below

• smolorg / smolgrad
  small auto-grad engine inspired from Karpathy's micrograd and PyTorch
  ☆276Updated 10 months ago
• naklecha / llm-inference-optimizations-explained
  in this repository, i'm going to implement increasingly complex llm inference optimizations
  ☆68Updated 4 months ago
• spikedoanz / from-bits-to-intelligence
  could we make an ml stack in 100,000 lines of code?
  ☆46Updated last year
• Laz4rz / GPT-2
  Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish
  ☆172Updated last year
• Maharshi-Pandya / cudacodes
  Learnings and programs related to CUDA
  ☆418Updated 2 months ago
• seatedro / arxival
  ☆93Updated 8 months ago
• smolorg / smolar
  a tiny multidimensional array implementation in C similar to numpy, but only one file.
  ☆227Updated last year
• arpitingle / gpu-alpha
  High Quality Resources on GPU Programming/Architecture
  ☆588Updated last year
• mesozoic-egg / tinygrad-notes
  Tutorials on tinygrad
  ☆414Updated last month
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆638Updated 2 weeks ago
• Quentin-Anthony / torch-profiling-tutorial
  ☆498Updated last month
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆410Updated 6 months ago
• smolorg / smolvecstore
  a tiny vectorstore implementation built with numpy.
  ☆63Updated last year
• 0xD4rky / Vision-Transformers
  This repo has all the basic things you'll need in-order to understand complete vision transformer architecture and its various implementa…
  ☆227Updated 8 months ago
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆194Updated 3 months ago
• kmohan321 / Research_Papers
  ☆46Updated 5 months ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆139Updated last year
• tokenbender / avataRL
  rl from zero pretrain, can it be done? yes.
  ☆269Updated this week
• omkaark / simple-federated-learning
  ☆96Updated last year
• spikedoanz / tensor-tic-tac-toe
  parallelized hyperdimensional tictactoe
  ☆125Updated last year
• nano-R1 / resources
  Compiling useful links, papers, benchmarks, ideas, etc.
  ☆45Updated 6 months ago
• obadakhalili / tinygrad-tensor-puzzles
  Solve puzzles to improve your tinygrad skills!
  ☆143Updated 6 months ago
• tugot17 / pmpp
  Complete solutions to the Programming Massively Parallel Processors Edition 4
  ☆522Updated 3 months ago
• adam-maj / adammaj.com
  adammaj.com | Built with Next & Chakra
  ☆90Updated last month
• EurekaLabsAI / tensor
  The Tensor (or Array)
  ☆446Updated last year
• dwarkeshsp / spaced-rep
  ☆56Updated 6 months ago
• PrimeIntellect-ai / prime-environments
  Training-Ready RL Environments + Evals
  ☆111Updated this week
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆279Updated last month
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆142Updated last year
• 1y33 / 100Days
  GPU Kernels
  ☆194Updated 5 months ago