Systems ML Engineer - AI Compiler & Kernel Optimizations

Meta Careers; Santa Clara, Northern (CA, KY); Hybrid; USD 180,000 - 240,000; Full time

Job summary

Meta's Santa Clara-based MTIA Software Team is hiring a Software Engineer, Systems ML specializing in compilers and kernels. You will help develop the AI compiler stack, contribute to PyTorch core components, and optimize high-performance kernels for next-generation hardware accelerators.

Collaborating with AI researchers and hardware engineers, you will analyze models, implement compiler optimizations, and advance deployment velocity for training and inference on MTIA platforms.

Responsibilities

• Develop SW stack for AI compiler, frameworks, and high-performance kernel development.
• Contribute to PyTorch core compilers to support new state-of-the-art AI hardware accelerators and optimize their performance.
• Analyze deep learning networks, develop & implement compiler optimization algorithms.

Job description

Meta's Santa Clara-based MTIA Software Team is hiring a Software Engineer, Systems ML specializing in compilers and kernels. You will help develop the AI compiler stack, contribute to PyTorch core components, and optimize high-performance kernels for next-generation hardware accelerators.

Collaborating with AI researchers and hardware engineers, you will analyze models, implement compiler optimizations, and advance deployment velocity for training and inference on MTIA platforms.