Machine Learning Optimization Specialist

6 days ago


Mountain View, California, United States MatX Full time
About MatX

We are pushing the boundaries of artificial intelligence at MatX by creating efficient AI models that harness the capabilities of physics.

Our mission is to develop comprehensive, vertically integrated full-stack solutions that span from silicon to systems, encompassing both hardware and software components. These solutions are designed to handle large machine learning workloads for advanced generative intelligence.

Job Description

This role involves building accurate performance models to guide scheduling decisions for existing and future machine learning models.

You will create high-quality libraries for efficient distributed training and serving.

Additionally, you will work closely with architects, hardware, and software teams to drive solutions from model development to deployment on metal.

Requirements
  • Bachelor's or Master's degree in Computer Science, Engineering, or a related field.
  • Strong programming skills in Python.
  • Expertise in machine learning frameworks such as JAX, PyTorch, or TensorFlow.
  • Deep understanding of the Transformer architecture.
  • Experience with distributed computing, high-performance networking, or large-scale machine learning systems.
Preferred Skills
  • Hands-on experience with flash attention, quantization, pruning, or other system 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 machine learning systems research through publication and/or industrial practice.
  • Experiences in pre-silicon exploration, post-silicon bringup, and fleet debug.
Salary and Benefits

The estimated annual salary for this position is $200,000, plus equity and benefits. We value diversity and inclusivity and are committed to equal employment opportunity.

This role requires authorization to work in the United States and may involve working from our offices in Mountain View on Tuesdays-Thursdays.

Benefits include:

  • A competitive salary range of $120,000 - $400,000.
  • Equity participation in the company.
  • A comprehensive benefits package.


  • Mountain View, California, United States Waymo Full time

    At Waymo, we're pushing the boundaries of autonomous driving technology. Our mission is to make transportation safer, more efficient, and accessible to everyone.We're looking for a highly skilled Machine Learning (ML) engineer to join our team. As an ML Engineer at Waymo, you'll play a key role in developing and optimizing our ML systems for compute...


  • Mountain View, California, United States Tik Tok Full time

    Job SummaryThe Machine Learning Infrastructure Specialist will be responsible for designing and implementing the infrastructure for TikTok's machine learning models. This role requires expertise in distributed systems, data engineering, and cloud computing.Key ResponsibilitiesDesign and develop scalable data pipelines for machine learning model training and...


  • Mountain View, California, United States Bonfy Full time

    About the Role: As a Senior software Engineer at Bonfy.AI, you will play a crucial role in developing, deploying, and optimizing machine learning models that scale across diverse applications. This role requires a deep foundation in software development, algorithms, and data science mathematics. You will work on cutting-edge technologies and directly impact...


  • Mountain View, California, United States Moveworks Full time

    Job DescriptionWe are seeking a highly skilled Senior Machine Learning Infrastructure Specialist to join our team at Moveworks. As a critical member of our AI infrastructure team, you will play a key role in building and optimizing cutting-edge machine learning systems for large language models.In this position, you will work closely with our...


  • Mountain View, California, United States Moveworks Full time

    At Moveworks, we're revolutionizing the way businesses interact with AI. As a Senior Machine Learning Infrastructure Specialist, you'll play a critical role in building and scaling our cutting-edge ML infrastructure.">We're looking for an expert in machine learning who can design, build, and optimize scalable ML infrastructure to support training,...


  • Mountain View, California, United States CV Library Full time

    Job OverviewWe are seeking a highly skilled Cloud Infrastructure Specialist to join our team at CV Library. This is a 12+ month contract opportunity that requires expertise in machine learning infrastructure, cloud platforms, and containerization technologies.Key Responsibilities:Design and implement scalable machine learning infrastructure on Google Cloud...


  • Mountain View, California, United States AnchorFree Full time

    About Us:AnchorFree is a fast-growing technology company in Silicon Valley that makes a significant impact on people's lives around the globe by enabling free access to all information and content online.Job Description:Develop and optimize new ways to present additional content to VPN users while they are browsing the WebApply state-of-the-art machine...


  • Mountain View, California, United States AnchorFree Full time

    Become a Key Player in Machine Learning DevelopmentWe are seeking an exceptional Data Scientist/Machine Learning Expert/AI Engineer to join our team at AnchorFree, a leading technology company in Silicon Valley.About Our Company:We provide a free service to users worldwide, enabling them to access the internet securely and privately.We have over 9 million...


  • Mountain View, California, United States Intuit Full time

    In this exciting role as a Predictive Modeling Expert, you'll work closely with our data science team to develop and deploy predictive models that drive business growth. Key responsibilities include:Main ResponsibilitiesDevelop Data-Driven Solutions: Create and refine features from underlying data, build pipelines to train and deploy models, and partner with...


  • Mountain View, California, United States AnchorFree Full time

    Job OpportunityWe are seeking a skilled Machine Learning Engineer to join our team at AnchorFree in Silicon Valley. As a fast-growing technology company, we enable free access to all information and content online, securely and privately.The successful candidate will be responsible for developing algorithms that increase the relevancy of content offered to...


  • Mountain View, California, United States Waymo Full time

    **Job Description**In this position, you will be responsible for designing, implementing, and optimizing the distributed training infrastructure for our machine learning models. You will work closely with cross-functional teams to develop solutions that improve the scalability, reliability, and performance of our ML frameworks.


  • Mountain View, California, United States MatX Full time

    About MatX\MatX's mission is to push the boundaries of artificial intelligence by developing hardware that enables efficient and scalable model training. We are a cutting-edge technology company that seeks to revolutionize the field of machine learning.\Job Summary\We are looking for a highly experienced Machine Learning Lead who can lead our team of...


  • Mountain View, California, United States Bonfy Full time

    Senior Software Engineer at Bonfy.AIBonfy.AI is a pioneering AI company that's changing the game. We're committed to making AI trustworthy and efficient. This role offers a unique opportunity to work on cutting-edge technologies and drive innovations that impact our growth and success.Key Responsibilities:Design, implement, and optimize machine learning...


  • Mountain View, California, United States Applied Intuition Full time

    About the RoleWe are seeking a skilled software engineer to join our team in developing and applying cutting-edge machine learning approaches to sensor simulation. In this role, you will collaborate with our research team to integrate state-of-the-art techniques for modeling worlds and sensors such as Lidars, Radars, and Cameras.Your primary focus will be on...


  • Mountain View, California, United States MatX Full time

    About the RoleWe are looking for a talented Machine Learning Hardware Engineer to join our team. The ideal candidate will have a deep understanding of Machine Learning Accelerator architectures and experience with benchmarking and optimizing code.Key ResponsibilitiesDesign and develop hardware models for our ML workloads.Collaborate with cross-functional...


  • Mountain View, California, United States Bonfy Full time

    About Bonfy.AIWe're a pioneering company that's working behind the scenes on revolutionary innovations to make AI more trustworthy. We're seeking a talented Senior Software Engineer to help us drive these advancements.Job OverviewThis role requires a deep foundation in software development, algorithms, and data science mathematics. You'll work on...


  • Mountain View, California, United States Waymo Full time

    Job TitleSenior Machine Learning Engineer, TrainingAbout the RoleWe are seeking a Senior Machine Learning Engineer to join our Hybrid team at Waymo. As a key member of our ML Infrastructure team, you will be responsible for developing the infrastructure components necessary for distributed training, implementing automation solutions for provisioning,...


  • Mountain View, California, United States Nuro Full time

    **About Us**">Nuro is a robotics company that aims to improve everyday life through innovative technologies. Founded in 2016, we have spent years developing autonomous driving (AD) technology and commercializing AD applications. Our world-class autonomous driving system, the Nuro DriverTM, combines AD hardware with our AI-first self-driving software.We've...


  • Mountain View, California, United States Samsung Electronics America North America Full time

    Samsung Electronics America North America is a leading advanced advertising technology company in rapid growth that focuses on enabling brands to connect with Samsung TV audiences as they are exposed to digital media by using the industry's most comprehensive data to build the world's smartest advertising platform.As part of an international organization...


  • Mountain View, California, United States Nuro Full time

    Nuro is a pioneering robotics company dedicated to enhancing everyday life through innovative technology. Founded in 2016, we have spent years developing autonomous driving (AD) solutions and commercializing AD applications. Our world-class Nuro DriverTM combines AD hardware with AI-first self-driving software, built to learn and improve through data.We've...