Modular is hiring a
AI GPU Performance Engineer

Summary

Join Modular as an AI GPU Performance Engineer to work with leading experts on performance libraries for GPUs, understand, analyze, profile, and optimize AI workloads on state-of-the-art hardware and software platforms.

Requirements

• Deep understanding of computer architecture (memory hierarchies, caching, etc.) and their impact on algorithm design
• 3+ years of relevant experience working on complex code and software systems
• Self motivated and independent with the ability to execute on agreed upon specifications
• Experience with GPU programming languages such as CUDA or OpenCL

Responsibilities

• Design, develop, and optimize high-performance AI numeric and data manipulation kernels/operators for GPUs
• Achieve state-of-the-art performance by leveraging software and micro-architectural features of GPUs
• Work with compiler, framework, runtime, and serving teams to deliver end-to-end performance that fully utilizes GPU workstations and servers
• Collaborate with machine learning researchers to guide system development for future ML trends

Benefits

• Premier insurance plans
• Up to 5% 401k matching
• Very generous WFH setup stipends
• Flexible paid time off