This is a fully remote position, open to applicants in Canada.

📋 Description

• Define, design, develop, test, debug, release, enhance, and maintain software tools tailored for domain-specific programming languages.

• Collaborate closely with hardware design teams, compiler development firms, and programming communities.

• Engage in language and standards groups.

• Design and implement both front-end and back-end components of the Triton compiler.

• Create optimization passes for supported kernels.

⛳️ Requirements

• A Bachelor's degree in Computer Science, Computer Engineering, or a related discipline with 4+ years of experience in GPU programming for AI workloads, OR a Master's degree in the same fields with 2+ years of experience, OR a PhD in the same fields with 1+ year of experience in GPU programming for AI workloads.

• Proficient programming skills in C, C++, and Python.

• Experience with various compiler stages, code generation, and optimization strategies.

• Familiarity with the GitHub development environment.

• Knowledge of PyTorch attention mechanisms for transformer models.

• Preferred Qualifications: Experience with LLVM, MLIR, and the Triton compiler architecture.