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

Meta · 4 days ago

ML Compiler and Performance Engineer

Meta is a technology company that builds solutions to connect people and grow businesses. The ML Compiler and Performance Engineer will be responsible for optimizing AI models' performance during their transition to MTIA chips, focusing on multi-node compute and addressing performance bottlenecks.
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 runtimesMasters degreeIn-domain experience

Required

Experience developing and deploying optimizations at the level of PyTorch/Aten or comparable stacks

Preferred

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

Benefits

Bonus
Equity
Benefits

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