ML Frameworks Engineer — Inference & Accelerator Systems

Meta; Bellevue (WA); On-site; USD 122,000 - 181,000; Full time

Job summary

Meta in Bellevue, WA seeks a Software Engineer for Inference Enablement within the PyTorch AI framework ecosystem. You will help build the foundational technology enabling state-of-the-art models across Ads, Reels, Feed, and GenAI, with hardware acceleration on CPU, GPU, and silicon.

The role emphasizes development of AI frameworks, compilers, and high-performance kernels, with opportunities to optimize model architectures and contribute to cutting-edge inference and training tooling.

Qualifications

• Python and C/C programming skills.
• Experience in AI framework development or accelerating deep learning models on hardware architectures.
• Must obtain work authorization in country of employment at the time of hire and maintain ongoing work authorization during employment.

Responsibilities

• Develop software stack focusing on AI frameworks, compiler stack, high-performance kernel development and acceleration for next-generation AI accelerators.
• Contribute to PyTorch core compilers to support new hardware accelerators and optimize performance.
• Analyze neural networks and implement compiler optimization algorithms.
• Accelerate next-gen deep learning models (recommendation systems, GenAI, CV, NLP).
• Performance tuning of deep learning framework and software components.

Skills

Python programming

C/C++ programming

AI framework development

Hardware acceleration

Education

BSc in CS/EE or related

Tools

Triton

CUDA

Job description

Meta in Bellevue, WA seeks a Software Engineer for Inference Enablement within the PyTorch AI framework ecosystem. You will help build the foundational technology enabling state-of-the-art models across Ads, Reels, Feed, and GenAI, with hardware acceleration on CPU, GPU, and silicon.

The role emphasizes development of AI frameworks, compilers, and high-performance kernels, with opportunities to optimize model architectures and contribute to cutting-edge inference and training tooling.