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 8 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 5 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 2 months ago
- Course on Flash-attention in Triton☆100Feb 9, 2026Updated 6 months ago
- ☆19Mar 3, 2025Updated last year
- Companion code for The Physics of LLM Inference book☆26Apr 21, 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.
- ☆19Mar 29, 2026Updated 4 months ago
- we have ai at home☆118Jun 18, 2026Updated last month
- ☆22Jul 20, 2026Updated 3 weeks ago
- coding CUDA everyday!☆77Feb 5, 2026Updated 6 months ago
- Pure Triton kernels for Qwen3.5-27B inference on NVIDIA B200☆120Feb 28, 2026Updated 5 months ago
- A self-study for CUDA principles from 0 to 1☆18Jan 23, 2026Updated 6 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☆218May 27, 2026Updated 2 months ago
- a personal collection of my notes for ml sys☆118Updated this week
- 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.
- a LLM inference engine to run on consumer hardware☆47Apr 15, 2026Updated 3 months ago
- ☆132Dec 9, 2025Updated 8 months ago
- mHC kernels implemented in CUDA☆265Mar 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…☆154Jun 10, 2026Updated 2 months ago
- ☆44May 4, 2025Updated last year
- 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
- SQL-like query language and CLI for Qdrant vector search engine☆46Jun 13, 2026Updated last month
- Hand-Rolled GPU communications library☆96Nov 25, 2025Updated 8 months ago
- documenting my work in inference engineering☆25Apr 19, 2026Updated 3 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- An educational distributed training and inference library for neural nets using local computing☆74Jun 10, 2026Updated 2 months ago
- A compact implementation of SGLang, designed to demystify the complexities of modern LLM serving systems.☆4,720May 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]☆48Apr 14, 2026Updated 3 months ago
- Educational WIP☆73Feb 16, 2026Updated 5 months ago
- An extension of the nanoGPT repository for training small MOE models.☆283Mar 9, 2025Updated last year
- ☆35Nov 11, 2025Updated 8 months ago
- ☆385May 4, 2026Updated 3 months ago
- MetaAttention: A Unified and Performant Attention Framework Across Hardware Backends(PPoPP'26)☆17Updated this week
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A repository consisting of paper/architecture replications of classic/SOTA AI/ML papers in pytorch☆424Nov 11, 2025Updated 8 months ago
- ☆177May 29, 2026Updated 2 months ago
- 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
- Implement a reasoning LLM in PyTorch from scratch, step by step☆4,948Updated this week
- dLLM: Simple Diffusion Language Modeling☆2,665Jul 17, 2026Updated 3 weeks ago
- ☆109Oct 16, 2025Updated 9 months ago