Software Engineer, Systems ML

Meta is looking for software engineers to play a pivotal role designed to further enhance and elevate our AI training and inference infrastructure. As a cornerstone of our team, you will play a significant role in refining and fine-tuning the performance of our AI models and in building user facing APIs for our ML engineers. Your expertise will enable us to reach new heights in model training and inference. The position requires a combination of expertise in machine learning and software engineering.

Software Engineer, Systems ML Responsibilities

• Analyze performance and power efficiency of deep learning inference workloads
• Build user facing frameworks for efficient training and inference
• Identify and prototype opportunities for performance optimization
• Partner with teams across meta reality labs to optimize key training and inference workloads.

Minimum Qualifications

• Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience.
• Specialized experience in one or more of the following machine learning/deep learning domains:, GPU architecture, ML systems, AI infrastructure, high performance computing, performance optimizations, or Machine learning frameworks (e.g. PyTorch), numerics
• Experience developing AI-System infrastructure or AI algorithms in C/C++ or Python

Preferred Qualifications

• Master/PhD degree in Computer Science, Computer Engineering
• 5+ years of experience in ML framework development or accelerating deep learning models on hardware architectures
• Experience with distributed systems development
• Optimizing machine learning model inference and training specifically on NVIDIA GPUs hardware.
• Experience with CUDA.
• Extensive experience in profiling GPU performance and distributed training.
• Deep understanding of Open AI Triton and on techniques for mixed precision training.

Start preparing

Learn about how to prepare for your interview with our interview guide, tips, and interactive experiences.

​Data Center

About Meta

Meta builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps like Messenger, Instagram and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D screens toward immersive experiences like augmented and virtual reality to help build the next evolution in social technology. People who choose to build their careers by building with us at Meta help shape a future that will take us beyond what digital connection makes possible today—beyond the constraints of screens, the limits of distance, and even the rules of physics.

Meta is committed to providing reasonable support (called accommodations) in our recruiting processes for candidates with disabilities, long term conditions, mental health conditions or sincerely held religious beliefs, or who are neurodivergent or require pregnancy-related support.

$172,994/year to $241,000/year + bonus + equity + benefits

Individual pay is determined by skills, qualifications, experience, and location. Compensation details listed in this posting reflect the base salary only, and do not include bonus, equity or sales incentives, if applicable. In addition to base salary, Meta offers benefits. Learn more about benefits at Meta.