Machine Learning Engineer

3 weeks ago


Seattle, United States Amazon Full time

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators and the Trn1 and Inf1 servers that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron.

This role is responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive scale large language models like Llama2, GPT2, GPT3 and beyond, as well as stable diffusion, Vision Transformers and many more.

The ML Apps team works side by side with compiler engineers and runtime engineers to create, build and tune distributed inference solutions with Trn1. Experience optimizing inference performance for both latency and throughput on these large models using Python, Pytorch or JAX is a must. Deepspeed and other distributed inference libraries are central to this and extending all of this for the Neuron based system is key.

Key job responsibilities

This role will help lead the efforts building distributed inference support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will help tune these models to ensure highest performance and maximize the efficiency of them running on the customer AWS Trainium and Inferentia silicon and the TRn1, Inf1 servers. Strong software development using C++/Python and ML knowledge are both critical to this role.

A day in the life

As you design and code solutions to help our team drive efficiencies in software architecture, you’ll create metrics, implement automation and other improvements, and resolve the root cause of software defects. You’ll also:

  1. Build high-impact solutions to deliver to our large customer base.
  2. Participate in design discussions, code review, and communicate with internal and external stakeholders.
  3. Work cross-functionally to help drive business decisions with your technical input.
  4. Work in a startup-like development environment, where you’re always working on the most important stuff.
About the team

Our team is dedicated to supporting new members. We have a broad mix of experience levels and tenures, and we’re building an environment that celebrates knowledge-sharing and mentorship. Our senior members enjoy one-on-one mentoring and thorough, but kind, code reviews. We care about your career growth and strive to assign projects that help our team members develop your engineering expertise so you feel empowered to take on more complex tasks in the future.

BASIC QUALIFICATIONS

- 3+ years of non-internship professional software development experience
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language

PREFERRED QUALIFICATIONS

- 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Bachelor's degree in computer science or equivalent

Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or other legally protected status.

#J-18808-Ljbffr

  • Seattle, United States Apple Full time

    Machine Learning Engineer, Applied Machine LearningSeattle, Washington, United StatesCorporate FunctionsDo you want to work on building groundbreaking ML technologies and make real world impact? At Apple, great ideas have a way of becoming great products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no...


  • Seattle, United States Apple Inc. Full time

    Machine Learning Engineer, Applied Machine LearningDo you want to work on building groundbreaking ML technologies and make real world impact? At Apple, great ideas have a way of becoming great products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish. The Applied...


  • Seattle, United States Aquent Full time

    Our client's team seeks a Machine Learning Engineer to advance the development and improvement of our software foundation and tools vital for training state-of-the-art AI models. Your role will be centered on creating strong, scalable, and efficient training infrastructures and frameworks facilitating the full spectrum of the machine learning process, from...


  • Seattle, United States Tubi Tv Full time

    Tubi is a global entertainment company and the most watched free TV and movie streaming service in the U.S. and Canada. Dedicated to providing all people access to all the world’s stories, Tubi offers the largest collection of on-demand content, including over 250,000 premium movies and TV episodes and over 300 exclusive originals. With a passionate...


  • Seattle, United States Tiger Analytics Full time

    Tiger Analytics is an advanced analytics consulting firm. We are the trusted analytics partner for several Fortune 100 companies, enabling them to generate business value from data. Our consultants bring deep expertise in Data Science, Machine Learning, and AI. Our business value and leadership have been recognized by various market research firms, including...


  • Seattle, United States Jupiter Power Full time

    Tiger Analytics is an advanced analytics consulting firm. We are the trusted analytics partner for several Fortune 100 companies, enabling them to generate business value from data. Our consultants bring deep expertise in Data Science, Machine Learning, and AI. Our business value and leadership have been recognized by various market research firms, including...


  • Seattle, United States PitchBook Full time

    At PitchBook, we are always looking forward. We continue to innovate, evolve, and invest in ourselves to bring out the best in everyone. We're deeply collaborative and thrive on the excitement, energy, and fun that reverberates throughout the company. Our extensive learning programs and mentorship opportunities help us create a culture of curiosity that...


  • Seattle, United States Aquent Full time

    Our client’s team seeks a Machine Learning Engineer to advance the development and improvement of our software foundation and tools vital for training state-of-the-art AI models. Your role will be centered on creating strong, scalable, and efficient training infrastructures and frameworks facilitating the full spectrum of the machine learning process, from...


  • Seattle, United States Expedia , Inc. Full time

    Expedia, Inc. has openings for Machine Learning Engineers (all types/levels) in Seattle, WA. Some positions may allow for telecommuting. Job Duties: Build and maintain machine learning pipeline end-to-end in production environments including engineering, model training, model scoring, and model validation. Design and develop state-of-the-art AI platform...


  • Seattle, United States Rec Room Full time

    Job DescriptionJob DescriptionRec Room is the best place to build and play games together. Players can chat, hang out, play in millions of rooms, or build something new to share with the world! We are looking for an exceptional Machine Learning Engineer to support our players & creators are we continue to grow. As a Machine Learning Engineer, you'll be...


  • Seattle, United States Amazon Full time

    Job ID: 2839145 | Amazon.com Services LLC - A57Are you looking to work at the forefront of Machine Learning and AI? Would you be excited to apply cutting edge Generative AI, ML and DL algorithms to solve real world problems with significant impact?Amazon is revolutionizing customer service yet again now with cutting edge AI and ML. Device, Digital and Alexa...


  • Seattle, United States Qualtrics Full time

    Qualtrics, LLC seeks Machine Learning Engineer in Seattle, WA. Duties: Work with Research Engineers to implement, tune, and productize machine learning models. Develop scalable, fast, robust, and simple web-based software solutions to solve complex business problems. Modify existing software to correct errors, allow it to adapt to new hardware, or to...


  • Seattle, Washington, United States Adobe Full time

    The OpportunityAt Adobe, we're looking for a talented Machine Learning intern to join our team and contribute to the development of innovative AI and machine learning solutions. As a Machine Learning intern, you will work on applying AI and machine learning techniques to big-data problems to help Adobe better understand, lead, and optimize the experience of...


  • Seattle, Washington, United States Tik Tok Full time

    About the Role:TikTok is a leading destination for short-form mobile video, inspiring creativity and bringing joy to users worldwide. As a key member of our Ecommerce Alliance Team, you will play a vital role in serving merchants and creators, driving business growth, and providing a personalized shopping experience for our users.We are seeking talented...

  • Machine Learning Engineer

    21 minutes ago


    Seattle, United States Newton Colmore Consulting Full time

    Machine Learning Engineer - Drug DiscoveryNewton Colmore is working with a discovery research company in Seattle at the forefront of innovation, assisting them with their search for a Machine Learning Engineer.The company is committed to developing solutions that address unmet medical needs and improve patient outcomes, utilizing advanced machine learning...


  • Seattle, United States Axon Full time

    Your Impact Are you eager to contribute to building AI products impacting the world for the greater good?As a ML Engineer for Embedded AI at Axon, you will contribute to developing AI solutions transforming the public safety space and supporting our company’s mission: accelerate justice, protect truth, and save lives. Working closely with other ML...


  • Seattle, Washington, United States Tik Tok Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Engineer to join our E-Commerce Risk Control (ECRC) team at TikTok. As a key member of our team, you will be responsible for developing and implementing state-of-the-art machine learning algorithms to prevent and detect risks in our e-commerce platforms.Key Responsibilities:Invent,...


  • Seattle, United States Newton Colmore Consulting Ltd Full time

    Newton Colmore is working with a discovery research company in Seattle at the forefront of innovation, assisting them with their search for a Machine Learning Engineer.The company are committed to developing solutions that address unmet medical needs and improve patient outcomes, utilising advanced machine learning algorithms.On behalf of our client, we are...


  • Seattle, Washington, United States Tik Tok Full time

    About the RoleTikTok is a leading platform for short-form mobile video, and our mission is to inspire creativity and bring joy. The Supply Chain and Logistics - Seller Growth team is responsible for leading efforts in merchant and creator acquisition, fostering the growth of the platform, and implementing incubation programs and incentives to support the...


  • Seattle, Washington, United States DoorDash Full time

    About the RoleWe're seeking a talented Applied Machine Learning expert to drive innovation and growth at DoorDash. As a key member of our team, you will leverage our robust data and machine learning infrastructure to build cutting-edge recommendation systems and implement new AI solutions to expand restaurant selection and drive business growth.You'll be...