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 last year
• Laz4rz / GPT-2
  Following Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish
  ☆172Updated last year
• naklecha / llm-inference-optimizations-explained
  in this repository, i'm going to implement increasingly complex llm inference optimizations
  ☆75Updated 6 months ago
• Quentin-Anthony / torch-profiling-tutorial
  ☆532Updated 4 months ago
• spikedoanz / from-bits-to-intelligence
  could we make an ml stack in 100,000 lines of code?
  ☆46Updated last year
• tokenbender / avataRL
  rl from zero pretrain, can it be done? yes.
  ☆282Updated 2 months ago
• obadakhalili / tinygrad-tensor-puzzles
  Solve puzzles to improve your tinygrad skills!
  ☆167Updated 2 months ago
• mesozoic-egg / tinygrad-notes
  Tutorials on tinygrad
  ☆443Updated 2 months ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆140Updated last year
• Maharshi-Pandya / cudacodes
  Learnings and programs related to CUDA
  ☆428Updated 5 months ago
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆662Updated this week
• NousResearch / atropos
  Atropos is a Language Model Reinforcement Learning Environments framework for collecting and evaluating LLM trajectories through diverse …
  ☆768Updated this week
• soketlabs / coom
  A training framework for large-scale language models based on Megatron-Core, the COOM Training Framework is designed to efficiently handl…
  ☆24Updated last month
• dwarkeshsp / spaced-rep
  ☆57Updated 8 months ago
• IST-DASLab / llmq
  Quantized LLM training in pure CUDA/C++.
  ☆221Updated this week
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆195Updated 6 months ago
• arpitingle / gpu-alpha
  High Quality Resources on GPU Programming/Architecture
  ☆589Updated last year
• seatedro / arxival
  ☆94Updated 10 months ago
• divyamakkar0 / JAXformer
  A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
  ☆105Updated 2 months ago
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆441Updated 9 months ago
• wlxiong / learn-ml-the-hard-way
  Learn machine learning the hard way
  ☆29Updated 5 years ago
• PrimeIntellect-ai / prime-environments
  Curated collection of community environments
  ☆195Updated this week
• Noumena-Network / nmoe
  MoE training for Me and You and maybe other people
  ☆239Updated this week
• smolorg / smolvecstore
  a tiny vectorstore implementation built with numpy.
  ☆63Updated last year
• KhoomeiK / sanskrit-ocr
  ☆45Updated 5 months ago
• 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 11 months ago
• pyember / ember
  ☆234Updated 5 months ago
• seatedro / glimpse
  Copy code from your codebase to clipboard instantly for LLM context!
  ☆341Updated 4 months ago
• groq / openbench
  Provider-agnostic, open-source evaluation infrastructure for language models
  ☆690Updated last week
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆153Updated 2 years ago