Lead AI Performance Engineer, Portable HPC (Remote)

Massachusetts Institute of Technology; Cambridge (MA); Hybrid; USD 120,000 - 160,000; Full time

Job summary

A prestigious educational institution in Cambridge, MA is seeking a Lead Research Software Engineer to enhance AI performance engineering. This role involves optimizing workloads for AMD GPUs and requires expertise in Python, C++, and AI frameworks like PyTorch. Candidates should have at least five years of experience in technical fields and demonstrate excellent communication skills. Remote work options are available for this two-year term position.

Qualifications

• Minimum of five years of experience in technical fields or computational research.
• Deep familiarity with AI/ML frameworks such as PyTorch, TensorFlow, or JAX.
• Experience with performance profiling and benchmarking on Linux systems.

Responsibilities

• Lead applied performance engineering for AI workloads.
• Optimize existing workloads for AMD GPUs.
• Profile AI workloads to identify performance bottlenecks.

Skills

Proficiency in Python

Proficiency in C++

Familiarity with AI/ML frameworks

Hands-on experience with GPU programming models

Performance profiling skills

Excellent communication skills

Self-motivation

Collaboration skills

Education

Bachelor’s degree or equivalent

Tools

NVIDIA GPUs

AMD MI355X GPUs

AMD ROCm software stack

Linux-based High-Performance Computing systems

Job description

A prestigious educational institution in Cambridge, MA is seeking a Lead Research Software Engineer to enhance AI performance engineering. This role involves optimizing workloads for AMD GPUs and requires expertise in Python, C++, and AI frameworks like PyTorch. Candidates should have at least five years of experience in technical fields and demonstrate excellent communication skills. Remote work options are available for this two-year term position.