Current jobs related to ML Compiler Team Lead - Menlo Park, California - META

  • Technical Lead for ML Compiler Development

    2 days ago

    Menlo Park, California, United States META Full time

    Company Overview:The PyTorch Compiler team at Meta is dedicated to advancing the performance and efficiency of PyTorch without compromising its flexibility and ease of use. Our team is instrumental in driving PT2, a groundbreaking step forward in PyTorch's history that integrated compiler technologies into its core. Since the initial release of PT2...

  • Senior Compiler Developer for Artificial Intelligence

    16 hours ago

    Menlo Park, California, United States META Full time

    As a Senior Compiler Developer for Artificial Intelligence at Meta, you will play a pivotal role in shaping the future of machine learning (ML) compilers. Our PyTorch Compiler team aims to push the boundaries of what's possible with AI, making it faster, more efficient, and accessible to everyone.Job Description:We are seeking an exceptional individual with...

  • Software Engineering Manager, PyTorch Compiler

    4 weeks ago

    Menlo Park, California, 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 Engineering Manager, AI Compiler

    2 days ago

    Menlo Park, California, United States The Meta Full time

    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 Engineering...

  • Software Engineer, Systems ML

    4 weeks ago

    Menlo Park, California, 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...

  • Software Engineering Manager, PyTorch Compiler

    3 days ago

    Menlo Park, California, United States The Meta Full time

    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 development...

  • Software Engineer, Systems ML

    4 weeks ago

    Menlo Park, California, 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...

  • Lead ML Scientist

    24 hours ago

    Menlo Park, California, United States Snowflake Computing Full time

    Build the future of the AI Data Cloud. Join the Snowflake team. There is only one Data Cloud. Snowflake's founders started from scratch and designed a data platform built for the cloud that is effective, affordable, and accessible to all data users. But it didn't stop there. They engineered Snowflake to power the Data Cloud, where thousands of organizations...

  • Research Scientist Intern, Systems ML

    4 weeks ago

    Menlo Park, California, United States Meta Inc Full time

    Summary:You will be a member of the Triton Accel team and part of the bigger industry-leading PyTorch AI framework organization. Triton is a breakthrough Python-based language and compiler for authoring performant kernels for AI Accelerators such as GPUs.Our team works on all aspects of Triton - compiler optimizations (architecture-independent and...

  • Systems ML Engineer

    2 days ago

    Menlo Park, California, United States META Full time

    Responsibilities:This role requires the development of the PT2 compiler, specifically working on TorchDynamo, TorchInductor, Export, and PyTorch Core. A crucial part of this job is improving PyTorch performance through systematic solutions that benefit the entire community.Develop the PT2 compiler (TorchDynamo, TorchInductor, Export, PyTorch Core).Improve...

  • ML Systems Engineer

    4 hours ago

    Menlo Park, California, United States META Full time

    Job DescriptionThis is an exciting opportunity to contribute to the development of state-of-the-art GenAI models. We are looking for a highly skilled ML Systems Engineer who can develop custom/novel architectures, define use cases, and develop methodology & benchmarks to evaluate different approaches.The ideal candidate will have experience working with...

  • Research Scientist Intern, Systems ML

    1 week ago

    Menlo Park, California, United States META Full time

    Research Scientist Intern, Systems ML - Frameworks / Compilers / Kernels (PhD) You will be a member of the Triton Accel team and part of the bigger industry-leading PyTorch AI framework organization. Triton is a breakthrough Python-based language and compiler for authoring performant kernels for AI Accelerators such as GPUs. Our team works on all aspects of...

  • Research Scientist Intern, Systems ML

    4 weeks ago

    Menlo Park, California, United States Meta Inc Full time

    You will be a member of the Triton Accel team and part of the bigger industry-leading PyTorch AI framework organization. Triton is a breakthrough Python-based language and compiler for authoring performant kernels for AI Accelerators such as GPUs. Our team works on all aspects of Triton - compiler optimizations (architecture-independent and dependent),...

  • ML Kernel and Compiler Optimization Expert

    2 days ago

    Menlo Park, California, United States META Full time

    **Job Requirements**We are looking for a highly skilled Research Engineer to join our team at META. As a key member of our Research & Development team, you will be responsible for developing and implementing various optimization techniques at the algorithm, kernel/compiler, and runtime levels.You will work closely with AI research scientists to accelerate...

  • Software Engineer, Systems ML

    4 weeks ago

    Menlo Park, California, United States Meta Inc Full time

    Summary:In this role, you will be a member of the PyTorch Accelerator Enablement(AE) team and part of the bigger industry-leading PyTorch AI framework organization. Our team focus on optimizing GenAI model inference on GPUs. As a key member of our Research & Development team, you will work on developing and implementing various optimization techniques at the...

  • PyTorch Compiler Specialist

    2 days ago

    Menlo Park, California, United States META Full time

    Join Our TeamAs a Deep Learning Acceleration Expert at META, you'll be part of a dynamic team focused on developing and optimizing AI frameworks for real-world applications. We're looking for someone with extensive experience in designing and developing high-performance software solutions for AI frameworks and compiler stacks. Your expertise in analyzing and...

  • Research Scientist, Systems ML and HPC

    4 days ago

    Menlo Park, California, United States Meta Inc Full time

    Research Scientist, Systems ML and HPC - SW/HW Co-DesignMeta is seeking a Research Scientist to join our Research & Development teams. The ideal candidate will have industry experience working on AI Infrastructure related topics. The position will involve taking these skills and applying them to solve crucial and exciting problems in the hardware/software...

  • Research Scientist, Systems ML

    2 weeks ago

    Menlo Park, California, United States META Full time

    Research Scientist, Systems ML - SW/HW Co-Design - InferenceMeta is seeking a Research Scientist to join our Research & Development teams. The ideal candidate will have industry experience working on AI Infrastructure related topics. The position will involve applying these skills to solve crucial problems on the web. We are hiring in multiple locations. The...

  • Software Engineer, Systems ML

    2 days ago

    Menlo Park, California, United States The Meta Full time

    Software Engineer, Systems ML - SW/HW Co-designApply to this job Location pin icon Sunnyvale, CA • Menlo Park, CA • Seattle, WA • Fremont, CA • Remote, US + 4 more - Hide Apply to this job Meta is seeking an AI Software Engineer to join our Research & Development teams. The ideal candidate will have industry experience working on AI Infrastructure...

  • Software Engineer, Systems ML

    2 days ago

    Menlo Park, California, United States The Meta Full time

    Meta is seeking an AI Software Engineer to join our Research & Development teams. The ideal candidate will have industry experience working on AI Infrastructure related topics. The position will involve taking these skills and applying them to solve for some of the most crucial & exciting problems that exist on the web. We are hiring in multiple...

ML Compiler Team Lead

2 weeks ago

Menlo Park, California, United States META Full time

**Job Summary**

We are seeking an experienced leader to join our PyTorch compiler team as a Senior Engineering Manager. In this role, you will be responsible for driving the development of PT2 technologies and leading a team of highly passionate and innovative compiler specialists.

**Key Responsibilities**

  • Support a team of engineers in a cutting-edge domain.
  • Lead with a strong sense of direction and stay informed about industry trends.
  • Bootstrap new initiatives across teams.
  • Drive business outcomes and maintain a healthy information channel.
  • Adapt to the team's needs and wear different hats as necessary.

**Requirements**

  1. Experience with PyTorch, ML runtime, or ML compilers.
  2. 2+ years of experience managing a team of ML framework, runtime, or compiler engineers.
  3. Leadership experience by setting directions, driving strategies, and establishing cross-functional collaborations.
  4. Demonstrated experience in recruiting, structuring, and leading technical organizations.

**Preferred Qualifications**

  1. Knowledge of or experience with PT2 technologies.
  2. Experience with other ML compilers such as XLA, TVM, ONNX, Halide, and Triton.
  3. Experience in vertical performance optimization, GPU performance, quantization, or distributed training.
  4. Knowledge of ML model architecture research landscape.