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
  ☆277Updated 11 months ago
• naklecha / llm-inference-optimizations-explained
  in this repository, i'm going to implement increasingly complex llm inference optimizations
  ☆70Updated 5 months ago
• 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
• Quentin-Anthony / torch-profiling-tutorial
  ☆523Updated 3 months ago
• spikedoanz / from-bits-to-intelligence
  could we make an ml stack in 100,000 lines of code?
  ☆46Updated last year
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆139Updated last year
• obadakhalili / tinygrad-tensor-puzzles
  Solve puzzles to improve your tinygrad skills!
  ☆146Updated 3 weeks ago
• Maharshi-Pandya / cudacodes
  Learnings and programs related to CUDA
  ☆422Updated 4 months ago
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆653Updated this week
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆195Updated 5 months ago
• seatedro / arxival
  ☆92Updated 9 months ago
• tokenbender / avataRL
  rl from zero pretrain, can it be done? yes.
  ☆280Updated last month
• smolorg / smolvecstore
  a tiny vectorstore implementation built with numpy.
  ☆63Updated last year
• arpitingle / gpu-alpha
  High Quality Resources on GPU Programming/Architecture
  ☆590Updated last year
• mesozoic-egg / tinygrad-notes
  Tutorials on tinygrad
  ☆436Updated 3 weeks ago
• PrimeIntellect-ai / pi-quant
  SIMD quantization kernels
  ☆91Updated 2 months ago
• dwarkeshsp / spaced-rep
  ☆56Updated 7 months ago
• IST-DASLab / llmq
  Quantized LLM training in pure CUDA/C++.
  ☆214Updated this week
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆147Updated 2 years ago
• PrimeIntellect-ai / prime-cli
  The Prime Intellect CLI provides a powerful command-line interface for managing GPU resources across various providers
  ☆102Updated this week
• kmohan321 / Research_Papers
  ☆46Updated 7 months ago
• NousResearch / atropos
  Atropos is a Language Model Reinforcement Learning Environments framework for collecting and evaluating LLM trajectories through diverse …
  ☆738Updated this week
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆287Updated 3 months ago
• spikedoanz / tensor-tic-tac-toe
  parallelized hyperdimensional tictactoe
  ☆125Updated last year
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆427Updated 7 months ago
• nuwandavek / karpathify
  ☆92Updated last year
• PrimeIntellect-ai / prime-environments
  Training-Ready RL Environments + Evals
  ☆164Updated this week
• omkaark / simple-federated-learning
  ☆96Updated last year
• VatsaDev / NanoPoor
  NanoGPT-speedrunning for the poor T4 enjoyers
  ☆72Updated 6 months ago
• tugot17 / pmpp
  Complete solutions to the Programming Massively Parallel Processors Edition 4
  ☆564Updated 4 months ago