Senior Machine Learning Compiler Engineer

Amazon Development Center U.S., Inc.
Kenmore, WA, United States
Full-time

The AWS Neuron Compiler team is seeking skilled compiler engineers to join our efforts in developing a state-of-the-art deep learning compiler stack for optimizing application models across diverse domains.

Your role will involve working with custom-built Machine Learning accelerators, including Inferentia / Trainium, to power solutions like Generative AI.

Key job responsibilities :

  • Develop and maintain tooling for the Neuron Compiler's accuracy and reliability
  • Lead the efforts building fuzzers and specification synthesis tooling for our LLVM-based compiler
  • Work in a team with a science focus to push the boundaries of what is known in delivering solutions to customers

Strong software development skills using C++ / Python are crucial. A science background in compiler development is strongly preferred, and experience in Machine Learning and AI accelerators is a plus.

The role is located in Seattle, Cupertino, Austin, or Toronto.

BASIC QUALIFICATIONS :

  • 6+ years of experience in developing compiler features and optimizations
  • Proficiency in C++ and Python programming applied to compiler or verification projects
  • Familiarity with LLVM, including knowledge of abstract interpretation and polyhedral domains
  • Demonstrated scientific approach to software engineering problems

PREFERRED QUALIFICATIONS :

  • Masters degree or PhD in computer science or equivalent
  • Experience with deep learning frameworks like TensorFlow or PyTorch
  • Understanding of large language model (LLM) training processes
  • Knowledge of CUDA programming for GPU acceleration
  • 1 day ago
Related jobs
Promoted
Amazon Development Center U.S., Inc.
Sammamish, Washington

The AWS Neuron Compiler team is seeking skilled compiler engineers to join our efforts in developing a state-of-the-art deep learning compiler stack for optimizing application models across diverse domains. A science background in compiler development is strongly preferred, and experience in Machine...

Promoted
FIS
Seattle, Washington

Senior Machine Learning Engineer. As a Machine Learning Data Engineer, you will tackle challenges ranging from identity theft to credit card fraud, to money laundering, and more. Our Fraud Intelligence team is on the cutting edge of data science and machine learning technology that detects and preve...

Promoted
Amazon
Seattle, Washington

Senior Business Intelligence Engineer, Amazon Ads. Are you a data-driven problem solver who thrives on uncovering strategic insights to drive business impact? If so, we have an exciting opportunity for you to join our centralized analytics team within Amazon Advertising as a Senior Business Intellig...

Promoted
Viant
Seattle, Washington

We're looking for a highly technical and mission-driven Senior Machine Learning Engineer to build real-time, low-latency machine learning services. The Machine Learning Engineering team builds Viant's proprietary machine learning system that will disrupt the Ad Tech industry. This individual will le...

Promoted
TikTok
Seattle, Washington

As a machine learning engineer on the Ads Signal team, you will develop novel machine learning solutions, build scalable tech foundations and launch various products to maximize signal values for ads in a privacy-preserving way. Responsible for developing machine learning and PET solutions for vario...

Promoted
Snap Inc
Seattle, Washington

We're looking for a Principal Machine Learning Engineer to join the Ad Ranking team at Snap!. Strong understanding of machine learning and deep learning approaches and algorithms, and their applications to advertising, recommendation, and/or search domain. Design, implement, and scale critical machi...

Promoted
TikTok
Seattle, Washington

Collaborate with engineering teams to deploy machine learning models, design online model workflow, and implement serving pipelines at scale. Work on state-of-the-art content understanding techniques including but not limited to unimodal/multimodal content classifiers, Active learning models, Reinfo...

Promoted
Apex Systems
Redmond, Washington

Position: Machine Learning Software Engineer. Apex is hiring a Machine Learning Software Developer to work with Researchers to take their ML models and optimize them. Candidates with strong C++, Python, and firmware experience who want experience with cutting-edge machine learning models, and electr...

Promoted
TikTok
Seattle, Washington

Utilizing cutting-edge machine learning technology, advanced NLP, CV, recommendation, and multi-modal technology, we're shaping a pioneering engine within the industry. Optimize the recommender system, focusing on hyperscale machine learning models. ...

DoorDash
Seattle, Washington

As a Machine Learning Engineer, you will have the opportunity to leverage our robust data and machine learning infrastructure to develop inference and ML models that impact millions of users across our three audiences and tackle our most challenging business problems. PhD or 3+ years of industry exp...