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 7 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.☆79Feb 18, 2026Updated 5 months ago
- A GPT-2 inference engine written from scratch in CUDA and C++. Implements custom CUDA kernels for tiled matrix multiplication, LayerNorm,…☆42May 17, 2026Updated 2 months ago
- Course on Flash-attention in Triton☆100Feb 9, 2026Updated 5 months ago
- ☆19Mar 3, 2025Updated last year
- Companion code for The Physics of LLM Inference book☆26Apr 21, 2026Updated 3 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.
- ☆15Mar 11, 2026Updated 4 months ago
- ☆19Mar 29, 2026Updated 3 months ago
- Qwen3-0.6B megakernel: 527 tok/s decode on RTX 3090 (3.8x faster than PyTorch)☆117Feb 10, 2026Updated 5 months ago
- we have ai at home☆117Jun 18, 2026Updated last month
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆19Feb 9, 2026Updated 5 months ago
- coding CUDA everyday!☆77Feb 5, 2026Updated 5 months ago
- Pure Triton kernels for Qwen3.5-27B inference on NVIDIA B200☆121Feb 28, 2026Updated 4 months ago
- a whirlwind tour to deep learning and deep learning systems☆81Updated this week
- 🌱 A little course on Reinforcement Learning Environments for evaluating and training Language Models☆216May 27, 2026Updated last month
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- a personal collection of my notes for ml sys☆107Updated this week
- a LLM inference engine to run on consumer hardware☆46Apr 15, 2026Updated 3 months ago
- ☆132Dec 9, 2025Updated 7 months ago
- mHC kernels implemented in CUDA☆264Mar 9, 2026Updated 4 months ago
- Winner 🏆 (Agent-only) MLSys 2026 - FlashInfer AI Kernel Generation Contest for the DeepSeek Sparse Attention (DSA) track with an average…☆148Jun 10, 2026Updated last month
- ☆44May 4, 2025Updated last year
- SQL-like query language and CLI for Qdrant vector search engine☆46Jun 13, 2026Updated last month
- Hand-Rolled GPU communications library☆96Nov 25, 2025Updated 7 months ago
- documenting my work in inference engineering☆25Apr 19, 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.
- An educational distributed training and inference library for neural nets using local computing☆72Jun 10, 2026Updated last month
- A compact implementation of SGLang, designed to demystify the complexities of modern LLM serving systems.☆4,616May 17, 2026Updated 2 months ago
- RVAA: Recursive Vision-Action Agent for Long Video Understanding. Implementation of the RLM paradigm (Zhang, Kraska, Khattab 2025)☆119Jan 15, 2026Updated 6 months ago
- Official PyTorch implementation for "TensorLens: End-to-End Transformer Analysis via High-Order Attention Tensors" [ACL 2026]☆47Apr 14, 2026Updated 3 months ago
- a fun and educational take on vLLM☆212Jan 25, 2026Updated 5 months ago
- Deep learning paper implementations☆19May 15, 2025Updated last year
- Educational WIP☆73Feb 16, 2026Updated 5 months ago
- An extension of the nanoGPT repository for training small MOE models.☆279Mar 9, 2025Updated last year
- ☆35Nov 11, 2025Updated 8 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.
- Personal CUDA learning repo, built step by step from scratch.☆95Jun 14, 2026Updated last month
- ☆380May 4, 2026Updated 2 months ago
- MetaAttention: A Unified and Performant Attention Framework Across Hardware Backends(PPoPP'26)☆16Dec 31, 2025Updated 6 months ago
- A repository consisting of paper/architecture replications of classic/SOTA AI/ML papers in pytorch☆425Nov 11, 2025Updated 8 months ago
- ☆174May 29, 2026Updated last month
- My submission for the GPUMODE/AMD fp8 mm challenge☆29Jun 4, 2025Updated last year
- Tutorials of Extending and importing TVM with CMAKE Include dependency.☆16Oct 11, 2024Updated last year