CSE 291 / DSC 291: Machine Learning Systems

This repository contains programming assignments for CSE 291 / DSC 291 in Spring 2026.

Current contents:

• pa1/CSE291-S26-PA1.ipynb: PA1 notebook and assignment writeup
• pa1/auto_diff.py: autodiff implementation scaffold
• pa1/transformer.py: transformer training and generation code
• pa1/tests: public tests
• pa2/README.md: PA2 assignment writeup (Triton matmul kernel + MPI tensor parallel)
• pa2/student_kernel.py: Triton kernel scaffold (Part 1)
• pa2/mpi_wrapper: comm.py scaffold for myAllreduce / myAlltoall (Part 2)
• pa2/model, pa2/data: tensor/data parallel scaffolds (Part 2)
• pa2/tests: public tests