rmsnorm
Here are 26 public repositories matching this topic...
249M-param MoE transformer built from scratch in PyTorch. GQA, RoPE, SwiGLU, sparse MoE with 3 aux losses, AMP training loop no Trainer abstractions. Architecture mirrors LLaMA/Mistral/Mixtral decisions, fully inspectable.
-
Updated
May 22, 2026 - Jupyter Notebook
Efficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.
-
Updated
Jun 5, 2024 - Python
Simple and easy to understand PyTorch implementation of Large Language Model (LLM) GPT and LLAMA from scratch with detailed steps. Implemented: Byte-Pair Tokenizer, Rotational Positional Embedding (RoPe), SwishGLU, RMSNorm, Mixture of Experts (MOE). Tested on Taylor Swift song lyrics dataset.
-
Updated
Nov 18, 2024 - Python
Fused Triton kernels for Transformer inference: RMSNorm+RoPE, Gated MLP, FP8 GEMM — CPU-testable references, autotuning, and benchmarking
-
Updated
May 25, 2026 - Python
Optimized Fused RMSNorm implementation with CUDA. Features vectorized memory access (float4), warp-level reductions, and efficient backward pass for LLM training
-
Updated
Dec 24, 2025 - Python
A non-official implementation of Qwen 3.5, as there doesn’t seem to be a paper or any code available that I can find, so I decided to implement it just for fun.
-
Updated
Mar 11, 2026 - Python
LLM pretraining from scratch on FineWeb dataset (architecture and all components explained), plus optimal use of GPU on SLURM cluster
-
Updated
May 12, 2026 - Python
A 36M-parameter goldfish language model with a 10-second memory + pixel-art PWA desk pet. Runs in your browser, fully offline. Adopt it at den-sec.github.io/glublm/desk-pet/
-
Updated
Jun 26, 2026 - JavaScript
Simple character level Transformer
-
Updated
May 27, 2024 - Jupyter Notebook
A CPU-friendly character-level GPT from scratch in PyTorch featuring modern architecture (RoPE, RMSNorm, GQA) and a modular RAG pipeline with Chroma DB.
-
Updated
Jul 1, 2026 - Python
A transformer language model built from scratch, from byte-level BPE tokenization through pretraining and QA fine-tuning.
-
Updated
May 25, 2026 - Python
Generative models nano version for fun. No STOA here, nano first.
-
Updated
Jun 10, 2026 - Jupyter Notebook
A from-scratch PyTorch LLM implementing Sparse Mixture-of-Experts (MoE) with Top-2 gating. Integrates modern Llama-3 components (RMSNorm, SwiGLU, RoPE, GQA) and a custom-coded Byte-Level BPE tokenizer. Pre-trained on a curated corpus of existential & dark philosophical literature.
-
Updated
Jan 7, 2026 - Python
Hands-on CUDA kernel engineering on LLM decode bottlenecks (RMSNorm, GEMV): PyTorch/Triton/CUDA C++ kernels profiled with Nsight and evaluated against a real vLLM backend.
-
Updated
Jun 28, 2026 - Python
Correctness-first Triton RMSNorm and SwiGLU kernels with cache-controlled GPU benchmarks and regression gates
-
Updated
Jun 14, 2026 - Python
Build an LLM in PyTorch: BPE tokenizer, GPT-1/2 + LLaMA, end-to-end train/infer
-
Updated
Mar 15, 2026 - Python
Improve this page
Add a description, image, and links to the rmsnorm topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the rmsnorm topic, visit your repo's landing page and select "manage topics."