Software Engineer, Systems ML

1 week ago


Menlo Park, United States META Full time

Software Engineer, Systems ML - PyTorch Compiler / ML Framework / Performance

Apply to this job

Location pin icon

Menlo Park, CA

Apply to this job

The PyTorch Compiler team is dedicated to making PyTorch run faster and more resource-efficient without sacrificing its flexibility and ease of use. The team is the driving force behind PT2, a step function change in PyTorch's history that brought compiler technologies to the core of PyTorch. PT2 technologies have gained industry-wide recognition since their first release in March 2023. The team is committed to building the PT2 compiler that withstands the test of time while striving to become the #1 ML framework compiler in the industry. The team is highly innovative, passionate about the technologies we build, and loves to do deep technical work. Our work is open-source, cutting-edge, and industry-leading.

Software Engineer, Systems ML - PyTorch Compiler / ML Framework / Performance Responsibilities

  • Develop the PT2 compiler (TorchDynamo, TorchInductor, Export, PyTorch Core).
  • Improve PyTorch performance via systematic solutions for the entire community.
  • Explore the intersection of the PyTorch compiler and PyTorch distributed.
  • Optimize Generative AI models across the stack (pre-training, fine-tuning, and inference).
  • Conduct cutting-edge research on ML compilers and ML distributed technologies.
  • Collaborate with users of PyTorch to enable new use cases of PT2 technologies both inside and outside Meta.
Minimum Qualifications
  • BS/MS in computer science or related field.
  • Research or industry experience in compilers, ML systems, ML accelerators, HPC, GPU performance, and similar.
  • Proficient in Python or CUDA programming
  • Currently has, or is in the process of obtaining a Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience. Degree must be completed prior to joining Meta.
Preferred Qualifications
  • Ph.D in computer science, applied mathematics, or related fields.
  • Familiarity with PT2 technologies, Triton, MLIR, or experiences working inside PyTorch.
  • Expert knowledge in GPU performance and writing high-performance CUDA kernels.
  • Research and software engineer experience demonstrated via fellowships, patents, internships, or coding competitions.
  • First-authored publications at peer-reviewed conferences (e.g., NeurIPS, MLSys, ASPLOS, PLDI, CGO, PACT, ICML, or similar).


For those who live in or expect to work from California if hired for this position, please click here for additional information.

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

Locations

Use ctrl + scroll to zoom the map

Zoom in

Zoom out

Recenter

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.

$56.25/hour to $173,000/year + bonus + equity + benefits

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

Equal Employment Opportunity and Affirmative Action

Meta is proud to be an Equal Employment Opportunity and Affirmative Action employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, reproductive health decisions, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, political views or activity, or other applicable legally protected characteristics. You may view our Equal Employment Opportunity notice here .

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. If you need support, please reach out to accommodations-ext@fb.com .

  • Menlo Park, United States META Full time

    Summary: The PyTorch Compiler team is dedicated to making PyTorch run faster and more resource-efficient without sacrificing its flexibility and ease of use. The team is the driving force behind PT2, a step function change in PyTorch’s history that brought compiler technologies to the core of PyTorch. PT2 technologies have gained industry-wide recognition...


  • Menlo Park, United States META Full time

    Software Engineer, Systems ML - PyTorch Compiler / ML Framework / Performance Apply to this job Location pin icon Menlo Park, CA Apply to this job The PyTorch Compiler team is dedicated to making PyTorch run faster and more resource-efficient without sacrificing its flexibility and ease of use. The team is the driving force behind PT2, a step function...


  • Menlo Park, United States META Full time

    Summary: The PyTorch Compiler team is dedicated to making PyTorch run faster and more resource-efficient without sacrificing its flexibility and ease of use. The team is the driving force behind PT2, a step function change in PyTorch’s history that brought compiler technologies to the core of PyTorch. PT2 technologies have gained industry-wide recognition...


  • Menlo Park, United States META Full time

    Summary: In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectures, combined...


  • Menlo Park, United States Character Full time

    About the Role As a Software Engineer you will work cross functionally with product, data science, data platform and ML researchers to identify, design and implement the applied ML infra, ranging from the data pipeline to model training and serving infrastructure, on one of the fastest growing consumer applications on the market today. You will lead...


  • Menlo Park, United States Lamini Full time

    Lamini enables every enterprise to safely, quickly, and cost-effectively build their own Expert AI. Our customers own their own models, trained on their data. Lamini optimizes for Expert AI workloads with minimal hallucination, enterprise-grade security, and enterprise flexibility, running on any infrastructure. Our team is made up of highly committed...

  • ML Research Engineer

    2 weeks ago


    Menlo Park, United States Diffuse Bio Full time

    The role: Extend and scale Diffuse's in-house deep generative modeling toolkit for    downstream applications in molecular design. Thoughtfully execute deep learning experiments to improve performance    of models or develop new functionality (e.g. loop engineering,    structure prediction of protein-protein complexes). Work closely with software...


  • Menlo Park, California, United States META Full time

    About METAMETA is a leader in the field of artificial intelligence and networking. Our mission is to create innovative technologies that enable fast and reliable communication between machines.Job DescriptionWe are seeking an experienced Software Engineering Manager to join our Network AI Software team. As a member of this team, you will be responsible for...

  • Software Engineer

    4 months ago


    Menlo Park, United States Diffuse Bio Full time

    The role: Design, build, and iterate on research infrastructure in close collaboration with research engineers. Build tools to automate and maintain computing clusters and data parsing pipelines.  Design and build software and APIs that enable internal and external access to our AI systems. Ideal background: Adaptability and openness to work on multiple...


  • Menlo Park, United States META Full time

    Summary: In this role, you will be a member of the Network AI Software team and part of the bigger DC networking organization. The team develops and owns the software stack around collective communication libraries around Meta.At the high level, the team aims to enable Meta-wide ML products and innovations to leverage our large-scale training and inference...


  • Menlo Park, CA, United States Lamini Full time

    Lamini enables every enterprise to safely, quickly, and cost-effectively build their own Expert AI. Our customers own their own models, trained on their data. Lamini optimizes for Expert AI workloads with minimal hallucination, enterprise-grade security, and enterprise flexibility, running on any infrastructure. Our team is made up of highly committed...


  • Menlo Park, United States Meta Inc Full time

    Summary: The MTIA (Meta Training & Inference Accelerator) Software team is part of AI Infra PyTorch org. The team’s mission is to explore, develop and help productize high-performance software and hardware technologies for AI at datacenter scale. The team co-optimizes both SW (e.g., algorithms and numerics) and HW (e.g., platform and network) to come up...


  • Allen Park, United States Apex Systems Full time

    Job#: 2052298 Job Description: Responsibilities: Design and develop AI systems for software quality and warranty. Lead the development and maintenance of AI and machine learning algorithms. Ensure the delivery of stable, high-quality infotainment products. Collaborate with various teams to meet product and engineering excellence. Thrive in a fast-paced,...


  • Allen Park, United States Apex Systems Full time

    Job#: 2052298 Job Description: Responsibilities: Design and develop AI systems for software quality and warranty. Lead the development and maintenance of AI and machine learning algorithms. Ensure the delivery of stable, high-quality infotainment products. Collaborate with various teams to meet product and engineering excellence. Thrive in a fast-paced,...


  • Menlo Park, United States META Full time

    Summary: The MTIA (Meta Training & Inference Accelerator) Software team has been developing a comprehensive AI Compiler strategy and optimizing compiler toolchains. This enables training and inference of Meta’s production DL/ML workloads on the specialized MTIA AI accelerator hardware in a highly performant and flexible way.We are looking for a Software...


  • Menlo Park, United States META Full time

    Summary: In this role, you will be a member of the Network AI Software team and part of the bigger DC networking organization. The team develops and owns the software stack around collective communication libraries around Meta.At the high level, the team aims to enable Meta-wide ML products and innovations to leverage our large-scale training and inference...


  • Menlo Park, United States Meta Inc Full time

    Summary: Meta Platforms, Inc. (Meta), formerly known as Facebook Inc., builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps and services like Messenger, Instagram, and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D...


  • Menlo Park, United States META Full time

    Summary: The PyTorch compiler team is the driving force behind PT2, a step function change in the history of PyTorch. PT2 technologies have gained industry-wide recognition since its first OSS release in 2023. But there is still a long and exciting technical roadmap going forward. We seek an Engineer Manager to drive subcomponents of the PyTorch Compiler...

  • Software Engineer

    5 days ago


    Menlo Park, United States Meta Inc Full time

    Software Engineer - PyTorch Domains Apply to this job Location pin icon Menlo Park, CA •New York, NY Apply to this job The PyTorch Domains team is at the forefront of machine learning innovation, community engagement and open-source development. We are dedicated to pushing the boundaries of ML technologies while fostering a vibrant, global community of...

  • Software Engineer

    1 month ago


    allen park, United States EPITEC Full time

    This role is for a Software Engineer who will work specifically on Infotainment systems, using Artificial Intelligence (AI) and Machine Learning (ML) to improve software quality and reliability. This position focuses on analyzing and predicting how infotainment systems will perform and ensuring they remain high-quality and stable over...