Software Engineer, Systems ML

3 weeks ago


Menlo Park, California, United States META Full time
Job Summary:

The PyTorch Compiler team at Meta is dedicated to making PyTorch run faster and more resource-efficient without sacrificing its flexibility and ease of use. We are seeking a highly skilled Software Engineer, Systems ML to join our team and contribute to the development of the PT2 compiler.

The ideal candidate will have a strong background in compilers, ML systems, and performance optimization. They will be responsible for developing the PT2 compiler, improving PyTorch performance, and exploring the intersection of the PyTorch compiler and PyTorch distributed.

We are looking for a team player who is passionate about building cutting-edge technologies and is committed to delivering high-quality results. If you are a motivated and talented engineer with a passion for ML and performance optimization, we encourage you to apply for this exciting opportunity.

Key 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

Requirements:

  • 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

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)

Compensation:

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

Industry:

Internet

Equal Opportunity:

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, or related medical conditions), sexual orientation, gender, gender identity, gender expression, transgender status, sexual stereotypes, age, status as a protected veteran, status as an individual with a disability, or other applicable legally protected characteristics.



  • Menlo Park, California, United States META Full time

    Job DescriptionMeta is seeking a highly skilled Research Scientist to join our Research & Development teams. The ideal candidate will have industry experience working on AI Infrastructure related topics and a strong background in Systems ML and HPC.Key ResponsibilitiesApply High-Performance Computing (HPC) algorithms and techniques to optimize large-scale AI...

  • Software Engineer

    4 weeks ago


    Menlo Park, California, United States Meta Full time

    Meta AI Software EngineerWe are seeking a highly skilled AI Software Engineer to join our Research & Development teams at Meta. As a key member of our team, you will be responsible for developing and applying AI and machine learning techniques to build intelligent language systems that improve our products and experiences.ResponsibilitiesApply relevant AI...


  • Menlo Park, California, United States META Full time

    Job Summary:In this role, you will be a key member of the Network AI Software team, part of the larger DC networking organization at Meta. The team is responsible for developing and owning the software stack around collective communication libraries.The team's primary goal is to enable Meta-wide ML products and innovations to leverage our large-scale...

  • Software Engineer IV

    4 weeks ago


    Menlo Park, California, United States BCforward Full time

    Job Title: Software Engineer IVBCforward is seeking a highly motivated Software Engineer IV for a Remote opportunity. The ideal candidate will have industry experience working on a range of recommendation, classification, and optimization problems. You will bring the ability to own the whole ML life cycle, define projects and drive excellence across...


  • Menlo Park, California, United States META Full time

    Job SummaryThe Meta AI Compiler Software team is seeking a Software Engineering Manager to lead the development and optimization of compiler toolchains for Meta's production DL/ML workloads on the MTIA AI accelerator hardware. The ideal candidate will have experience with compiler architecture, development, and management, as well as a strong understanding...

  • Software Engineer IV

    3 weeks ago


    Menlo Park, California, United States BCforward Full time

    About the Role:We are seeking a highly motivated Software Engineer IV to join our team at BCforward. As a key member of our engineering team, you will be responsible for designing, developing, and deploying large-scale software applications.Key Responsibilities:Adapt standard machine learning methods leveraging modern parallel environments (e.g. distributed...

  • Research Scientist

    3 weeks ago


    Menlo Park, California, United States META Full time

    Research Scientist - Systems ML and HPC Co-Design ExpertMeta is seeking a highly skilled Research Scientist to join our Research & Development teams. The ideal candidate will have industry experience working on AI Infrastructure related topics, with a strong focus on Systems ML and HPC Co-Design.Key Responsibilities:Apply High-Performance Computing (HPC)...


  • Menlo Park, California, United States META Full time

    Meta Hardware Systems EngineerMeta is seeking a skilled Hardware Systems Engineer to join our Release to Production (RTP) team. As a key member of this team, you will be responsible for the end-to-end Hardware Lifecycle of all Meta servers, including prototyping of experimental HW, pre-production hands-on system and hardware debugging and stress testing,...


  • Menlo Park, California, United States Robinhood Full time

    About the RoleWe are seeking a highly skilled Staff Software Engineer to join our Customer Care team, where you will play a pivotal role in shaping the future of our customer experience through cutting-edge generative AI technologies.As a Staff Software Engineer, you will partner closely with our AI/ML teams and other product teams to create AI-powered...


  • Menlo Park, California, United States Reconstruct Full time

    At Reconstruct, we are on a mission to become the visual reality 'digital twin' global standard for capital assets. Our Visual Command Center uses AI and ML to align & track what is there (reality) vs. what should be there (design) at any point in time (schedule). We are seeking a talented leader to head our software development team, driving the creation of...

  • Software Engineer

    1 week ago


    Menlo Park, California, United States Orca Bio Full time

    Job Title: Software EngineerAs a Software Engineer at our company, you will be responsible for developing and maintaining complex software systems. This includes designing, coding, testing, and deploying software applications. With a strong focus on collaboration and communication, you will work closely with cross-functional teams to deliver high-quality...


  • Menlo Park, California, United States META Full time

    About the Role:We're seeking a talented Software Engineer to join our team at Meta. As a key member of our engineering team, you'll have the opportunity to work on highly impactful technical challenges and help shape the future of our products.As a Software Engineer at Meta, you'll be responsible for designing, developing, and deploying high-quality software...


  • Menlo Park, California, United States Brio Digital Full time

    Job Title: Senior Lead Software EngineerAbout the Role:We are seeking a highly skilled Senior Lead Software Engineer to join our team at Brio Digital. As a Senior Lead Software Engineer, you will be responsible for owning the architecture, development, and deployment of high availability systems that empower AI Agents.Key Responsibilities:Design and develop...


  • Menlo Park, California, United States META Full time

    Meta DSP Software EngineerMeta is revolutionizing the way people connect and interact with technology. As a DSP Software Engineer, you will play a crucial role in developing software for consumer devices, including speakers, AR, and VR headsets. Your expertise in real-time audio algorithms and Android systems will enable us to create immersive experiences...


  • Menlo Park, California, United States Cyngn Full time

    About CyngnCyngn is a publicly traded autonomous vehicle company based in Menlo Park, CA. We have a culture of collaboration, diversity, and continuous learning. Our self-driving technology can be deployed in various commercial domains across various vehicle form factors.About the RoleWe are seeking a skilled Full Stack Engineer to contribute to the...


  • Menlo Park, California, United States META Full time

    Meta Production Engineer Job DescriptionMeta is building technologies that help people connect, find communities, and grow businesses. As a Production Engineer at Meta, you will play a critical role in developing, designing, creating, modifying, and testing software services to ensure optimal performance and capacity for growth.ResponsibilitiesDevelop and...


  • Menlo Park, California, United States META Full time

    About the Role:We're seeking a talented Software Engineer Intern to join our team at Meta. As a Software Engineer, you'll play a key role in developing the systems behind our products, creating web applications that reach billions of people, and building high-volume servers.Responsibilities:Develop and push production-ready code by quickly ramping on...


  • Menlo Park, California, United States META Full time

    Digital Signal Processing Software EngineerMeta is seeking a highly skilled Digital Signal Processing Software Engineer to join our team. As a key member of our software engineering team, you will be responsible for developing software for a variety of consumer devices, including speakers, AR and VR headsets. Your expertise in digital signal processing will...


  • Menlo Park, California, United States META Full time

    Production Systems Engineer, Fleet AI SystemsMeta is seeking a highly skilled Production Systems Engineer to join our Release to Production (RTP) team. As a key member of our team, you will be responsible for the Hardware Lifecycle of all Meta servers, including pre-production hands-on system and hardware debugging and stress testing, enabling...


  • Menlo Park, California, United States META Full time

    Job Title: Production Systems Engineer, Fleet AI SystemsMeta is seeking a highly skilled Production Systems Engineer to join our Release to Production (RTP) team. Our servers and data centers are the foundation upon which our rapidly scaling infrastructure operates efficiently to deliver our innovative services.Responsibilities:Interface with external...