Cloud Scale Machine Learning Developer

4 weeks ago


Cupertino, California, United States Amazon Full time

**Role Overview:**

We are seeking an experienced Software Development Engineer to join our Machine Learning Applications team for AWS Neuron. This role involves developing, enabling, and tuning machine learning models for cloud-scale applications.

The ideal candidate will have strong software development skills in C++/Python and ML knowledge, with experience in distributed inference libraries such as Deepspeed. They will work closely with compiler engineers and runtime engineers to optimize inference performance and develop distributed inference solutions with Trn1.

**Responsibilities:

  • Develop high-quality software components for machine learning models.
  • Collaborate with team members to design and implement distributed inference solutions.
  • Analyze and troubleshoot issues related to model performance and scalability.
  • Communicate technical information to both technical and non-technical audiences.

**Compensation and Benefits:

Amazon offers a competitive salary range of $129,300 - $223,600 per year, depending on location and experience. Additional benefits include medical, financial, and other perks.



  • Cupertino, California, United States Amazon Full time

    Job ResponsibilitiesDevelop, enable, and performance tune various machine learning models, including large language models.Optimize inference performance for latency and throughput using Python, PyTorch, or JAX.Work with our Machine Learning Applications team to create high-impact solutions that deliver exceptional results for our customers.Participate in...


  • Cupertino, California, United States Amazon Full time

    About the JobWe are seeking a skilled Cloud Scale Machine Learning Engineer to join our team. As a key member of our Machine Learning Applications team, you will be responsible for developing, enabling, and tuning distributed inference solutions using AWS Neuron.Salary: $173,450 - $178,650 per yearKey ResponsibilitiesDesign and implement high-performance...


  • Cupertino, California, United States Amazon Full time

    Job DescriptionThis role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. The team works side by side with compiler engineers and runtime engineers to create, build, and tune distributed inference solutions with Trn1.As a cloud-scale machine learning engineer, you will be responsible for optimizing inference...


  • Cupertino, California, United States Annapurna Labs Full time

    A comprehensive software stack for AWS's cloud-scale machine learning accelerators, designed to unlock the full potential of PyTorch and JAX frameworks. As a key member of the Annapurna Labs team, you'll be responsible for developing and enhancing support for these leading ML frameworks on the Trainium and Inferentia accelerators.Key...


  • Cupertino, California, United States Amazon Full time

    Cloud Scale Solutions Developer: We are seeking a talented developer to join our team in creating and optimizing cloud-scale machine learning solutions for AWS Neuron. This role involves working closely with compiler engineers and runtime engineers to build and tune distributed inference solutions.Main Responsibilities:Design and develop scalable machine...


  • Cupertino, California, United States Annapurna Labs Full time

    **Job Title:** Software Development Manager, AWS Neuron Machine Learning Distributed Training**Location:** Remote (with occasional travel to Amazon offices)**About Us:Annapurna Labs is a fast-paced and innovative company at the forefront of cloud computing. We're looking for a highly skilled software development manager to lead our team in designing and...


  • Cupertino, California, United States Apple Full time

    **Job Description:**We are seeking a highly experienced Senior Software Engineer to join our Machine Learning Platform Team. As a key member of this team, you will be responsible for designing and building cloud-native infrastructure platforms at Apple scale to support the deployment and operation of our AI/ML services and applications.Key...


  • Cupertino, California, United States Amazon Full time

    About the TeamWe are a dynamic team of experts in machine learning and software development, working together to create innovative solutions for cloud-scale inference. Our team is passionate about using technology to make a positive impact on society, and we are committed to fostering a culture of inclusivity, diversity, and respect. If you are a motivated...


  • Cupertino, California, United States Amazon Full time

    About Amazon">Amazon is a leader in cloud computing, artificial intelligence, and related technologies. We are committed to innovation and excellence, with a focus on developing cutting-edge solutions that improve people's lives.This role is part of our Machine Learning Applications (ML Apps) team, which works on developing and optimizing cloud-scale machine...


  • Cupertino, California, United States Apple Full time

    **Job Description**We are looking for a highly skilled Machine Learning Engineer, Developer Experience Specialist to join our team at Apple. In this role, you will work closely with our applied ML scientists and engineers to enhance the experience and productivity of software developers at Apple and in the Apple developer ecosystem.**About the Team**Our team...


  • Cupertino, California, United States Amazon Full time

    Job OverviewAmazon Web Services (AWS) is a leading provider of cloud-based services that power the world's most innovative businesses. We are seeking an experienced ASIC Design Engineer to join our Cloud-Scale Machine Learning Acceleration team, responsible for designing and optimizing hardware in our data centers.The successful candidate will have a deep...


  • Cupertino, California, United States Apple Full time

    **Job Summary**We are seeking a talented Senior Software Development Engineer, Machine Learning Expert to join our team at Apple. As a key member of our applied ML scientists and engineers team, you will be responsible for enhancing the experience and productivity of software developers at Apple and in the Apple developer ecosystem.**About the Role**In this...


  • Cupertino, California, United States Amazon Full time

    About the RoleWe are seeking a highly skilled Software Engineer to lead the development of machine learning tools for our cloud-scale accelerators. As a member of our team, you will design and implement new tools, pipelines, and automation to optimize system performance and ensure high availability and scalability.The ideal candidate will have experience...


  • Cupertino, California, United States Amazon Full time

    We are seeking an experienced ASIC Design Engineer to join our Cloud-Scale Machine Learning Acceleration team at Amazon. This is a challenging and rewarding opportunity for a skilled engineer to design and optimize hardware in our data centers, including AWS Inferentia, our custom designed machine learning inference datacenter server.About the RoleAs an ASIC...


  • Cupertino, California, United States Apple Full time

    We are seeking a highly skilled Senior AI Engineer to join our team at Apple, working on large-scale machine learning infrastructure.As a key member of our Foundation Model Infrastructure team within the Machine Learning Platform Technologies organization, you will play a critical role in building frameworks, services, and tools that power Apple's largest...


  • Cupertino, California, United States Amazon Full time

    ResponsibilitiesThe Cloud-Scale Machine Learning Acceleration team at AWS designs and optimizes custom chips and software stacks to accelerate innovation in the cloud. As an ASIC Design Engineer, your key responsibilities will include:Designing and optimizing hardware components for our machine learning serversCollaborating with software engineers to ensure...


  • Cupertino, California, United States Amazon Full time

    About the Role: Amazon's Machine Learning Engineering team is seeking a talented Team Lead to join our team. As a key member of our ML Apps team, you will be responsible for leading the development and deployment of large-scale machine learning models on AWS Neuron. This includes designing and implementing distributed training solutions using PyTorch,...


  • Cupertino, California, United States Apple Full time

    About the JobWe're looking for a highly skilled Data Science Manager to join our team at Apple. In this role, you will be responsible for leading the development of data-driven insights to inform product development and improvement. You will collaborate with machine learning and product engineering teams to drive product impact via data-driven insights.Key...


  • Cupertino, California, United States Apple Full time

    Company OverviewCupertino, California, United StatesSoftware and ServicesWe're a team of applied ML scientists and engineers who work to enhance the experience and productivity of software developers at Apple and in the Apple developer ecosystem. Our mission is to solve real-world problems using state-of-the-art ML models.Job DescriptionSalary: $175,800 -...


  • Cupertino, California, United States Amazon Full time

    Are you passionate about designing and developing cutting-edge machine learning solutions? Do you thrive in a fast-paced environment where innovation is key?About the RoleWe are seeking an experienced Machine Learning Architect to join our team at Amazon. As a key member of our ML Applications team, you will be responsible for developing and implementing...