Cloud Scalable Machine Learning Engineer

2 days ago


Seattle, Washington, United States Amazon Full time
About the Job

AWS Neuron is a cloud-scale software stack for AWS Inferentia and Trainium, our machine learning accelerators. We are looking for a Senior Distributed Training Specialist to join our team in Annapurna Labs, responsible for developing, enabling, and performance-tuning a wide variety of ML model families, including large language models like GPT and Llama.

The ideal candidate has experience with training massive-scale models using Python, as well as expertise in distributed training libraries such as FSDP and Deepspeed. This role will involve working closely with chip architects, compiler engineers, and runtime engineers to create and tune distributed training solutions with Trainium instances.

Key Responsibilities
  • Developing distributed training support into PyTorch and TensorFlow using XLA and the Neuron compiler and runtime stacks.
  • Tuning models to ensure highest performance and maximize efficiency on custom AWS Trainium and Inferentia silicon and Trn1, Inf1/2 servers.

Benefits
  • $151,300/year - $261,500/year base pay (depending on location).
  • Total compensation package includes equity, sign-on payments, and other benefits.
  • Flexible work environment with opportunities for career growth and development.

Requirements
  • Bachelor's degree in Computer Science or equivalent.
  • 5+ years of professional software development experience.
  • Experience with machine learning, data mining, information retrieval, statistics, or natural language processing.
  • Mastery of Python programming language.


  • Seattle, Washington, United States Amazon Full time

    Job OverviewAs a seasoned machine learning engineer, you will play a pivotal role in designing and implementing large-scale data structures for analytics and data science at Amazon.com Services LLC.With a strong background in machine learning techniques and software development experience, you will work closely with Applied Scientists, Data Scientists, and...


  • Seattle, Washington, United States Amazon Full time

    Job OverviewWe are seeking a highly skilled Senior Machine Learning Engineer to join our team at Amazon. This is an exciting opportunity for an experienced professional in machine learning and artificial intelligence to drive innovation and growth in our organization.About the RoleThe ideal candidate will have a strong background in machine learning, with a...


  • Seattle, Washington, United States ZipRecruiter Full time

    Job DescriptionWe are seeking a highly skilled Machine Learning Solutions Architect to join our team. In this role, you will play a pivotal part in designing and implementing our machine learning infrastructure to enhance and accelerate the development of our AI solutions.As a key member of our product development team, you will be responsible for developing...


  • Seattle, Washington, United States Tredence Inc. Full time

    As a Cloud Engineer Lead at Tredence Inc., you will play a key role in designing and implementing scalable data solutions on Google Cloud Platform (GCP). This is an exciting opportunity to leverage your expertise in cloud computing, big data, and machine learning to drive business growth and innovation.Tredence is a global analytics services and solutions...


  • Seattle, Washington, United States Amazon Full time

    Job DetailsThis Sr. Software Development Engineer role is part of the Sponsored Products Ad Response Prediction Team at Amazon Advertising. As a key member of this team, you will play a vital role in shaping the future of ad response prediction and machine learning at Amazon.We're seeking a highly skilled Sr. SDE ML to join our team and contribute to the...


  • Seattle, Washington, United States Apple Full time

    We are seeking a talented Machine Learning Platform Reliability Engineer to join our AI/ML Production Engineering team. In this role, you will be responsible for ensuring the reliability and scalability of our machine learning platform.You will work closely with cross-functional teams to identify and resolve issues, improve processes, and automate manual...


  • Seattle, Washington, United States Optimum Energy LLC Full time

    We are seeking an experienced Machine Learning Engineer to join our team at Optimum Energy LLC.Company Overview:Our company focuses on optimizing energy efficiency in buildings through innovative technology and predictive analytics.About the Role:This is a highly technical position that requires a strong background in machine learning, software development,...


  • Seattle, Washington, United States Moloco Full time

    About Us:Moloco is a cutting-edge machine learning company empowering organizations to grow and unlock the full value of their unique first-party data. We enable businesses to elevate their performance advertising capabilities, leveraging innovative technologies to drive scalable growth. Our mission is to empower businesses of all sizes to grow through...


  • Seattle, Washington, United States Aquent Full time

    Job OverviewAquent Talent is seeking a Machine Learning Engineer - AI Development to join our client's team in Redmond, Washington. As a key member of this team, you will be responsible for advancing the development and improvement of our software foundation and tools vital for training state-of-the-art AI models.Key Responsibilities:Create and uphold...


  • Seattle, Washington, United States Amazon Full time

    About the Role:Sponsored Products, a core product offering within Amazon's advertising portfolio, is seeking an Applied Scientist to drive end-to-end Machine Learning projects that address complex challenges in native advertising. As a key member of our Bespoke Shopping Experience team, you will develop customer-facing experiences and machine learning models...


  • Seattle, Washington, United States Metropolis Full time

    About the Role: Metropolis is an artificial intelligence company at the forefront of computer vision technology, enabling frictionless experiences in everyday life. Our mission is to revolutionize parking and beyond, providing unparalleled convenience and speed. We are seeking a talented Senior Machine Learning Engineer to join our growing team and...


  • Seattle, Washington, United States F5 Networks Full time

    **Transform Business Outcomes with Machine Learning Expertise at F5 Networks**We're seeking a highly skilled Software Engineer II to join our innovative technology team and leverage machine learning expertise to drive business growth and improve customer experiences.This is an exciting opportunity to work on cutting-edge projects, collaborate with...

  • Data Science Engineer

    3 weeks ago


    Seattle, Washington, United States Optimum Energy Full time

    Company OverviewOptimum Energy is a leading provider of software solutions that optimize energy consumption in buildings. Our innovative cloud-based platform and expert engineering support ensure significant electricity savings, water savings, and reduced carbon emissions.SalaryThe estimated annual salary for this role is $149,750, based on industry...


  • Seattle, Washington, United States Salesforce Full time

    About the RoleWe're looking for a highly skilled Data Scientist to join our team at Salesforce. As a Data Scientist, you will be responsible for building predictive models and recommender systems that drive personalized marketing and customer experience. You will work closely with cross-functional teams to design, develop, and deliver statistical models and...


  • Seattle, Washington, United States Amazon Full time

    We are seeking a highly skilled Business Intelligence and Machine Learning Engineer to join our team in Sponsored Products and Brands. As a key member of our Agile team, you will have the opportunity to drive a critical business area in Amazon Ads.Your responsibilities will include designing, developing, and deploying scalable offline machine-learning...


  • Seattle, Washington, United States Apple Full time

    Role SummaryWe are seeking an experienced Machine Learning Solutions Architect to join our Machine Learning Platform Technology (MLPT) team at Apple. As a key member of this team, you will play a pivotal role in translating the needs of our machine learning engineers into effective and scalable technical solutions.Key Responsibilities:Work closely with...


  • Seattle, Washington, United States Hispanic Technology Executive Council Full time

    OverviewSucceed in your career as a Cloud Engineering Lead at the Hispanic Technology Executive Council, where you will lead the design and implementation of scalable cloud solutions on Google Cloud Platform (GCP).About the RoleThis is a challenging opportunity to drive business growth through innovative cloud engineering practices, leveraging your expertise...


  • Seattle, Washington, United States Apple Full time

    About the PositionThis is an exciting opportunity to join Apple's Machine Learning Platform Technology (MLPT) team as a Solution Architect. As a key member of the team, you will work closely with machine learning engineers to design and implement scalable and efficient technical solutions.Job DetailsThe estimated annual salary for this position is between...


  • Seattle, Washington, United States Thomson Reuters Full time

    About the JobThis Machine Learning Engineer Intern role is an exciting opportunity to contribute to the incorporation of the next generation of Artificial Intelligence into Reuters News tooling. As an intern, you will work closely with our engineering team to integrate deep learning and machine learning models into cloud-based applications.You will be...


  • Seattle, Washington, United States Amazon Full time

    Job DescriptionAs a Software Development Engineer in Machine Learning at Amazon, you will drive the technical direction of our offerings and solutions, working with multiple technologies across the sponsored products organization.You will design, code, troubleshoot, and support scalable machine-learning pipelines and online serving systems. You will work...