Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Meta builds technologies that help people connect, find communities, and grow businesses through social platforms and immersive experiences.
$85,100 - $251,000
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
8+ years of experience
AI

Description For Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Meta is seeking a Software Engineer for their MTIA (Meta Training & Inference Accelerator) Software team within the PyTorch AI framework organization. This role focuses on developing AI compiler strategies and high-performance solutions for ML workloads. You'll work on PyTorch framework components, AI compiler and runtime, and high-performance kernels for MTIA AI hardware platforms. The position involves close collaboration with AI researchers and hardware design teams to optimize deep learning models and develop compiler optimizations. You'll be at the forefront of machine learning infrastructure, working with state-of-the-art AI technologies including Recommendation systems, Generative AI, Computer Vision, and NLP. The role offers competitive compensation, including equity and benefits, and the opportunity to shape the future of AI infrastructure at one of the world's leading tech companies. The ideal candidate should have strong C/C++ programming skills and experience with AI frameworks or hardware acceleration. This is an excellent opportunity for experienced engineers passionate about AI infrastructure and performance optimization to make a significant impact on Meta's AI capabilities.

Last updated 8 days ago

Responsibilities For Software Engineer, Systems ML - Frameworks / Compilers / Kernels

  • Development of SW stack with focus on AI frameworks, compiler stack, high performance kernel development and acceleration
  • Contribute to the development of PyTorch AI framework core compilers
  • Analyze deep learning networks, develop & implement compiler optimization algorithms
  • Collaborate with AI research scientists to accelerate deep learning models
  • Performance tuning and optimizations of deep learning framework & software components

Requirements For Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Python
  • Proven C/C++ programming skills
  • Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
  • Experience in AI framework development or accelerating deep learning models on hardware architectures

Benefits For Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Equity
Medical Insurance
  • Equity
  • Medical Insurance

Interested in this job?

Jobs Related To Meta Software Engineer, Systems ML - Frameworks / Compilers / Kernels

AI Research Engineer, Computer Vision - FAIR

Senior AI Research Engineer position at Meta FAIR, focusing on computer vision research and development of foundation models for Meta's products.

Software Engineer, Systems ML - SW/HW Co-design

Senior ML Software Engineer role at Meta focusing on systems and hardware co-design for AI infrastructure optimization.

Software Engineer, Machine Learning

Senior Machine Learning Engineer role at Meta, developing scalable ML solutions and leading technical teams in London, focusing on recommendation systems and classification problems.

Machine Learning ASIC Engineer, Architecture

Senior Machine Learning ASIC Engineer position at Meta focusing on developing and architecting machine learning accelerators for data center efficiency.

Software Engineer, Machine Learning

Senior Machine Learning Engineer role at Meta focusing on developing scalable ML solutions and leading technical teams.