A PyTorch implementation of the GPT-OSS-20B architecture. All components are coded from scratch: RoPE with YaRN, RMSNorm, SwiGLU with clamping and residual connection, Mixture-of-Experts (MoE), Self-Attention with learned sinks, banded attention, GQA, and KV-cache.
☆238Dec 2, 2025Updated 9 months ago
Alternatives and similar repositories for gpt-oss-20B
Users that are interested in gpt-oss-20B are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A series of high-performance GEMM (General Matrix Multiply) implementations Iteratively optimised for H100 GPUs in Pure CUDA.☆81Feb 18, 2026Updated 6 months ago
- A GPT-2 inference engine written from scratch in CUDA and C++. Implements custom CUDA kernels for tiled matrix multiplication, LayerNorm,…☆43May 17, 2026Updated 3 months ago
- Course on Flash-attention in Triton☆103Feb 9, 2026Updated 6 months ago
- ☆19Mar 3, 2025Updated last year
- Companion code for The Physics of LLM Inference book☆27Apr 21, 2026Updated 4 months ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- ☆19Mar 29, 2026Updated 5 months ago
- Qwen3-0.6B megakernel: 527 tok/s decode on RTX 3090 (3.8x faster than PyTorch)☆135Feb 10, 2026Updated 6 months ago
- we have ai at home☆118Jun 18, 2026Updated 2 months ago
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆19Feb 9, 2026Updated 6 months ago
- ☆22Jul 20, 2026Updated last month
- coding CUDA everyday!☆78Feb 5, 2026Updated 6 months ago
- Pure Triton kernels for Qwen3.5-27B inference on NVIDIA B200☆121Feb 28, 2026Updated 6 months ago
- a whirlwind tour to deep learning and deep learning systems☆82Updated this week
- 🌱 A little course on Reinforcement Learning Environments for evaluating and training Language Models☆224May 27, 2026Updated 3 months ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- a personal collection of my notes for ml sys☆122Updated this week
- ☆132Dec 9, 2025Updated 8 months ago
- a LLM inference engine to run on consumer hardware☆48Apr 15, 2026Updated 4 months ago
- mHC kernels implemented in CUDA☆266Mar 9, 2026Updated 5 months ago
- Winner 🏆 (Agent-only) MLSys 2026 - FlashInfer AI Kernel Generation Contest for the DeepSeek Sparse Attention (DSA) track with an average…☆158Aug 21, 2026Updated last week
- An efficient and scalable attention module designed to reduce memory usage and improve inference speed in large language models. Designe…☆26Jun 25, 2025Updated last year
- ☆45May 4, 2025Updated last year
- SQL-like query language and CLI for Qdrant vector search engine☆47Jun 13, 2026Updated 2 months ago
- Hand-Rolled GPU communications library☆95Nov 25, 2025Updated 9 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- documenting my work in inference engineering☆25Apr 19, 2026Updated 4 months ago
- An educational distributed training and inference library for neural nets using local computing☆77Jun 10, 2026Updated 2 months ago
- A compact implementation of SGLang, designed to demystify the complexities of modern LLM serving systems.☆4,935May 17, 2026Updated 3 months ago
- RVAA: Recursive Vision-Action Agent for Long Video Understanding. Implementation of the RLM paradigm (Zhang, Kraska, Khattab 2025)☆118Jan 15, 2026Updated 7 months ago
- Official PyTorch implementation for "TensorLens: End-to-End Transformer Analysis via High-Order Attention Tensors" [ACL 2026]☆49Apr 14, 2026Updated 4 months ago
- a fun and educational take on vLLM☆222Jan 25, 2026Updated 7 months ago
- Deep learning paper implementations☆20May 15, 2025Updated last year
- An extension of the nanoGPT repository for training small MOE models.☆285Mar 9, 2025Updated last year
- ☆35Nov 11, 2025Updated 9 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Personal CUDA learning repo, built step by step from scratch.☆104Jun 14, 2026Updated 2 months ago
- ☆387May 4, 2026Updated 3 months ago
- MetaAttention: A Unified and Performant Attention Framework Across Hardware Backends(PPoPP'26)☆17Aug 6, 2026Updated 3 weeks ago
- A repository consisting of paper/architecture replications of classic/SOTA AI/ML papers in pytorch☆425Nov 11, 2025Updated 9 months ago
- ☆178May 29, 2026Updated 3 months ago
- Tutorials of Extending and importing TVM with CMAKE Include dependency.☆16Oct 11, 2024Updated last year
- Implement a reasoning LLM in PyTorch from scratch, step by step☆5,137Updated this week