Systems for Machine Learning (CMSC828G)

Group Project

Overview

In the second half of the course, you will work on a group project in groups of two or three. The idea is to use the systems concepts learned in the course to accelerate and/or parallelize an ML workload. You can select the project topic yourself but you will have to discuss it with the instructors once to ensure that the scope is appropriate. An example project could be writing high performance Triton kernels for an existing ML workflow in your research. Another potential topic could be distributing a single GPU workload to multiple GPUs efficiently. Please reach out to the instructors if you need help coming up with project ideas.

Group Formation and Project Proposal

The first deadline is a project proposal and group formation. You should turn in a 1-2 page PDF with the following sections.

  • Group members: Names of the 2-3 group members.
  • Background and Motivation: Describe the systems or performance problem you want to solve.
  • Potential Approach(es): Describe your planned solution(s) to the problem. It's ok to only have rough ideas at this point.
  • Required Compute Estimate: An important part of systems for ML research is planning out your compute budget. Provide a rough estimate of the necessary compute for your project (CPU hours, GPU hours, etc.).

Interim Report

The interim report is a short report (2-3 page PDF) detailing progress so far. You should have preliminary profiling/performance results in your interim report.

Final Report

At the end of the semester, students will present their projects in class and submit a final report. The presentations will be in the last two weeks and each should be 12 minutes with 3-5 minutes for questions. The final report should be written like a research paper (introduction, background, methodology, results, etc...) in the NeurIPS format.

Important Dates
Topic Due on*
Group formation and project proposal Mar 4
Interim report Apr 17
Final presentation May 6-13
Final report and code May 15

*All deadlines are at 11:59 PM.