- Zurich, Zurich, Switzerland
- Internship · Intern
Responsibilities
- Design and implement Python DSL features for GPU kernel authoring, including metaprogramming and IR generation.
- Develop MLIR dialects, transformation passes, and lowering pipelines that preserve the information needed for effective GPU optimization.
- Design and optimize GPU kernels, exploring how IR design and compiler transformations affect generated code and performance.
- Profile, benchmark, and validate compiler and kernel improvements across representative workloads.
Requirements
- Currently pursuing or holding a Bachelor’s, Master’s, or Ph.D. degree in Computer Science, Computer Engineering, or a related technical discipline, or equivalent practical experience.
- Nice to have: CUTLASS
- Nice to have: CuTeDSL
- Nice to have: Triton
- Nice to have: LLVM