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
  ☆275Updated 10 months ago
• spikedoanz / from-bits-to-intelligence
  could we make an ml stack in 100,000 lines of code?
  ☆46Updated last year
• naklecha / llm-inference-optimizations-explained
  in this repository, i'm going to implement increasingly complex llm inference optimizations
  ☆68Updated 4 months ago
• Maharshi-Pandya / cudacodes
  Learnings and programs related to CUDA
  ☆420Updated 3 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
  ☆509Updated 2 months ago
• obadakhalili / tinygrad-tensor-puzzles
  Solve puzzles to improve your tinygrad skills!
  ☆145Updated this week
• seatedro / arxival
  ☆92Updated 8 months ago
• arpitingle / gpu-alpha
  High Quality Resources on GPU Programming/Architecture
  ☆589Updated last year
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆643Updated this week
• mesozoic-egg / tinygrad-notes
  Tutorials on tinygrad
  ☆419Updated last week
• 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…
  ☆228Updated 9 months ago
• smolorg / smolar
  a tiny multidimensional array implementation in C similar to numpy, but only one file.
  ☆228Updated last year
• NousResearch / atropos
  Atropos is a Language Model Reinforcement Learning Environments framework for collecting and evaluating LLM trajectories through diverse …
  ☆714Updated this week
• smolorg / smolvecstore
  a tiny vectorstore implementation built with numpy.
  ☆63Updated last year
• dwarkeshsp / spaced-rep
  ☆56Updated 6 months ago
• tokenbender / avataRL
  rl from zero pretrain, can it be done? yes.
  ☆275Updated 3 weeks ago
• seatedro / glimpse
  Copy code from your codebase to clipboard instantly for LLM context!
  ☆324Updated 2 months ago
• 8090-inc / top-coder-challenge
  go the distance
  ☆284Updated 4 months ago
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆421Updated 7 months ago
• Graffioh / bertos-inferno
  My path from leetcode hell to bigtech heaven
  ☆16Updated last year
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆139Updated last year
• adam-maj / robotics
  A deep dive on the history of robotics and the future of humanoids
  ☆108Updated 10 months ago
• geohotstan / tinycorp-meetings
  ☆94Updated last week
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆709Updated this week
• adam-maj / adammaj.com
  adammaj.com | Built with Next & Chakra
  ☆91Updated last week
• jam3scampbell / music
  control your music from the command line
  ☆143Updated 5 months ago
• IST-DASLab / llmq
  Quantized LLM training in pure CUDA/C++.
  ☆198Updated last week
• omkaark / simple-federated-learning
  ☆96Updated last year
• PrimeIntellect-ai / pi-quant
  SIMD quantization kernels
  ☆87Updated last month