1. Home
  2. Courses
  3. CS336
  4. Assignments
  5. Assignment 1

RMSNorm and SwiGLU

May 27, 2026

Cs336Cs336 Assignment 1

Implemented RMSNorm — pretty straightforward. Implemented SiLU, GLU, and SwiGLU. I think I did it right? Passed the tests, but not sure if I was supposed to compose the SwiGLU with the GLU more directly than I did.

rms_norm.py

positionwise_feedforward.py

By Mark Hudson
  • GitHub
  • Hugging Face
  • LinkedIn
  • Substack
  • X