ML Compiler and Performance Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Meta · 14 hours ago

ML Compiler and Performance Engineer

Meta builds technologies that help people connect, find communities, and grow businesses. In this role, you will have end-to-end responsibility for the performance of in-production AI models in their transition from stock HW to MTIA chips, focusing on models that require multi-node compute.

Computer Software
check
Comp. & Benefits

Responsibilities

Identifying bottlenecks and quantifying opportunities for improving performance
In-depth, end-to-end performance analysis and reporting
Developing optimizations to address identified bottlenecks
Optimizing compute/communication overlap
Work closely with other compiler teams as well as client teams (Recommendation Systems, Generative AI, etc)

Qualification

PyTorchMulti-node distributed computeOptimizing runtimesBachelor's degreeCommunication skills

Required

Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
Experience developing and deploying optimizations at the level of PyTorch/Aten or comparable stacks

Preferred

A Masters degree and 4+ years in-domain experience
A PhD degree and 2+ years in-domain experience
Experience optimizing multi-node distributed compute
Experience optimizing runtimes and/or kernels for accelerator platforms

Company

Meta's mission is to build the future of human connection and the technology that makes it possible.

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Kathryn Glickman
Director, CEO Communications
linkedin
leader-logo
Christine Lu
CTO Business Engineering NA
linkedin
Company data provided by crunchbase