Systems ML Engineer: AI Compiler & Kernel Innovator

Meta; New York (NY); On-site; USD 122,000 - 181,000; Full time

Benefits offered by this job

Bonus

Equity

Benefits

Job summary

Meta MTIA Software Team is building a comprehensive AI compiler and runtime platform to train and serve new DL/ML architectures, delivering high performance across hardware. You will work on PyTorch core compilers, AI frameworks, and high-performance kernels to accelerate workloads on MTIA hardware.

Collaboration with researchers will drive compiler optimizations for cutting-edge models. The role emphasizes performance tuning, tool development, and integration of AI workflows, with opportunities

Qualifications

• Currently has, or is in the process of obtaining a Bachelor's degree in CS/CE or related field; degree must be completed prior to joining Meta.
• Proven C/C++ programming skills.
• Experience in AI framework development or accelerating deep learning models on hardware architectures.
• Experience with PyTorch and related AI frameworks.

Responsibilities

• Development of the SW stack with AI compiler stack, frameworks, high-performance kernel development and acceleration onto hardware architectures.
• Contribute to the PyTorch AI framework core compilers to support new inference and training AI accelerators and optimize performance.
• Analyze deep learning networks, develop and implement compiler optimization algorithms.
• Collaborate with AI research scientists to accelerate next-gen DL models (recommendation, generative AI, CV, NLP).
• Performance tuning and optimizations of deep learning framework components.

Skills

C/C++ programming

AI framework development

GPU/CPU architectures

Education

Bachelor's degree in CS/CE or related field

Tools

PyTorch

Caffe2

TensorFlow

ONNX

TensorRT

Job description

Meta MTIA Software Team is building a comprehensive AI compiler and runtime platform to train and serve new DL/ML architectures, delivering high performance across hardware. You will work on PyTorch core compilers, AI frameworks, and high-performance kernels to accelerate workloads on MTIA hardware.

Collaboration with researchers will drive compiler optimizations for cutting-edge models. The role emphasizes performance tuning, tool development, and integration of AI workflows, with opportunities