Demo

ML Performance Engineer

Matx
Mountain View, CA Full Time
POSTED ON 1/16/2025
AVAILABLE BEFORE 4/13/2025

MatX's mission is to make the world's best AI models run as efficiently as allowed by physics, bringing the world years ahead in AI quality and availability. We are developing vertically integrated full-stack solutions from silicon to systems including hardware and software to train and run the largest ML workloads for AGI. We are looking for people who are excited about systems-focused ML research.

Responsibilities include :

  • Build performance models and tooling to validate and guide scheduling decisions for current and future ML models.
  • Write production-grade libraries for efficient distributed training and serving.
  • Collaborate with architects, hardware, and software teams to drive solutions from model to metal.

Requirements :

  • Bachelor's or Master's degree in Computer Science, Engineering, or a related field
  • Strong programming skills in Python
  • Expertise in ML frameworks such as JAX, PyTorch, or Tensorflow.
  • Deep knowledge of the Transformer architecture.
  • Experience with distributed computing, high performance networking, or large-scale ML systems,
  • Preferred Skills : Any of the following :

  • Hands on experience with flash attention, quantization, pruning, or other systems performance optimizations.
  • Fluency in parallelism strategies that balance computation, communication, and memory to improve throughput and latency of large models.
  • Experience with performance analysis tools and profilers for large scale systems.
  • Solid understanding of computer architecture and low-level optimization techniques
  • A track record of impactful ML Systems Research, through publication and / or industrial practice.
  • Experiences in pre-silicon exploration, post-silicon bringup, and fleet debug.
  • Compensation : The US base salary for this full-time position is $120,000 - $400,000 equity benefits

    As part of our dedication to the diversity of our team and our focus on creating an inviting and inclusive work experience, MatX is committed to a policy of Equal Employment Opportunity and will not discriminate against an applicant or employee on the basis of race, color, religion, creed, national origin or ancestry, sex, gender, gender identity, gender expression, sexual orientation, age, physical or mental disability, medical condition, marital / domestic partner status, military and veteran status, genetic information or any other legally recognized protected basis under federal, state or local laws, regulations or ordinances.

    All candidates must be authorized to work in the United States and work from our offices in Mountain View Tuesdays-Thursdays.

    This position requires access to information that is subject to U.S. export controls. This offer of employment is contingent upon the applicants capacity to perform job functions in compliance with U.S. export control laws without obtaining a license from U.S. export control authorities.

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a ML Performance Engineer?

    Sign up to receive alerts about other jobs on the ML Performance Engineer career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $97,257 - $120,701
    Income Estimation: 
    $123,167 - $152,295
    Income Estimation: 
    $59,440 - $93,329
    Income Estimation: 
    $69,043 - $113,369
    Income Estimation: 
    $70,609 - $91,165
    Income Estimation: 
    $86,680 - $110,316
    Income Estimation: 
    $117,033 - $148,289
    Income Estimation: 
    $86,680 - $110,316
    Income Estimation: 
    $110,730 - $135,754
    Income Estimation: 
    $117,033 - $148,289
    Income Estimation: 
    $110,730 - $135,754
    Income Estimation: 
    $128,617 - $162,576
    Income Estimation: 
    $117,033 - $148,289
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at Matx

    Matx
    Hired Organization Address Mountain View, CA Full Time
    MatX's mission is to make the world's best AI models run as efficiently as allowed by physics, bringing the world years ...
    Matx
    Hired Organization Address Mountain View, CA Full Time
    MatX's mission is to make the world's best AI models run as efficiently as allowed by physics, bringing the world years ...
    Matx
    Hired Organization Address Mountain View, CA Full Time
    MatX is on a mission to be the compute platform for AGI. We are developing vertically integrated full-stack solutions fr...
    Matx
    Hired Organization Address Mountain View, CA Full Time
    MatX is on a mission to make the world’s best AI models run as efficiently as possible, bringing the world years ahead i...

    Not the job you're looking for? Here are some other ML Performance Engineer jobs in the Mountain View, CA area that may be a better fit.

    Principal ML Performance Engineer

    Advanced Micro Devices, Inc, San Jose, CA

    Principal ML Training Performance Engineer

    Advanced Micro Devices, Inc, San Jose, CA

    AI Assistant is available now!

    Feel free to start your new journey!