Job Overview
Amazon Web Services (AWS) is seeking a Senior Machine Learning Engineer to join the AWS Neuron Apps team, focusing on ML Inference Applications. This role involves the development, enablement, and performance tuning of a wide variety of ML model families, including large language models like Llama2, GPT2, GPT3, stable diffusion, and Vision Transformers.
Responsibilities
- Lead the efforts in building distributed inference support into Pytorch and TensorFlow using XLA and the Neuron compiler and runtime stacks.
- Tune models to ensure highest performance on AWS Trainium and Inferentia silicon and TRn1, Inf1 servers.
- Engage in software development using C++/Python and apply ML knowledge to optimize inference performance for both latency and throughput.
Day-to-Day Activities
- Design and code solutions to drive efficiencies in software architecture.
- Create metrics, implement automation, and resolve software defects.
- Participate in design discussions, code reviews, and communicate with internal and external stakeholders.
- Work cross-functionally to help drive business decisions with your technical input.
About the Team
The ML Apps team works closely with compiler engineers and runtime engineers to create, build, and tune distributed inference solutions. The team environment supports new members through mentorship and knowledge-sharing, fostering an inclusive and collaborative work culture.
Career Growth and Development
AWS values diverse experiences and is committed to building a team that represents a variety of backgrounds, perspectives, and skills. The role offers significant opportunities for career growth, aiming to help team members develop into well-rounded professionals capable of taking on complex tasks in the future.
Work/Life Balance
AWS promotes work-life harmony by offering flexible work arrangements, ensuring that success at work does not come at the expense of your personal life.
Inclusive Team Culture
Our team embraces diversity and inclusion, actively fostering a workplace where everyone feels valued and inspired. We support various employee-led affinity groups and host ongoing learning experiences to promote an inclusive environment.
Benefits Extracted with AI
- Health insurance
- Paid vacation
- Flexible work hours
- Remote work options
Similar jobs
Last update: 23 minutes ago
Senior Systems Engineer, Managed Operations
Join AWS as a Senior Systems Engineer in Berlin to lead operations for the European Sovereign Cloud, ensuring high-availability AWS services.
Senior Software Engineer - AWS, Python, Ruby on Rails
Join HeyJobs as a Senior Software Engineer to design scalable systems using AWS, Python, and Ruby on Rails in a dynamic team.
Cloud Data Engineer
Seeking a Cloud Data Engineer with expertise in AWS, Python, and CI/CD for a hybrid role in Hannover. Join our dynamic team!
Senior Python Developer with AWS Experience
Join Basetime BV as a Senior Python Developer to develop and maintain AWS cloud solutions. Hybrid work, competitive salary, and growth opportunities.
Staff Software Engineer
Join Aiven as a Staff Software Engineer to develop cloud operations platforms using open-source technologies. Hybrid work in Berlin.
Senior Backend Engineer C++
Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.
Expert Machine Learning Engineer
Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.
Senior Solution Architect AWS (Freelance)
Seeking a Senior Solution Architect AWS for a freelance role in Frankfurt, Germany. Remote work available. Extensive AWS experience required.
Senior Backend Engineer - Java, Rust, Go
Join Together AI as a Senior Backend Engineer in Amsterdam. Work with Java, Rust, and Go to build scalable backend systems.
Senior Backend Engineer (Go)
Join Nebius as a Senior Backend Engineer (Go) to develop fault-tolerant cloud services in a hybrid work environment.
Senior Software Engineer (C++)
Join Nebius as a Senior Software Engineer (C++) to develop reliable cloud services in a hybrid work environment.
Software Engineer - Cloud Applications and Python
Join Topicus as a Software Engineer in Arnhem to develop cloud applications using Python, REST APIs, and ETL processes for healthcare data services.
Senior Software Engineer - Python, Apache Kafka
Join Aiven as a Senior Software Engineer in Berlin, focusing on Python and Apache Kafka in a hybrid work environment.
Software Engineer AWS & TypeScript
Join Haufe Akademie as a Software Engineer specializing in AWS & TypeScript to develop innovative cloud solutions in Freiburg.
Senior DevOps Engineer
Join CARFAX Europe as a Senior DevOps Engineer to manage AWS infrastructure, develop CI/CD pipelines, and enhance system observability.
Senior Software Engineer - LLM
Join Snowflake as a Senior Software Engineer to build scalable machine learning platforms with LLMs, leveraging Python and TensorFlow.
Senior Backend Developer
Join Motius as a Senior Backend Developer to work on cutting-edge R&D projects using AWS, Docker, GraphQL, and more in a hybrid work environment.
AI Engineer - Machine Learning and Robotics
Join Blueprint as an AI Engineer in Machine Learning and Robotics, focusing on scalable AI model training systems. Hybrid role in Redmond, WA.
Front End Development Engineer - AWS Commerce Platform
Join AWS as a Front End Development Engineer to build high-quality user interfaces using Angular, React, and TypeScript in Berlin.
Senior Software Developer - AWS, Java, Python, Scala
Join Omnia Retail as a Senior Software Developer, working with AWS, Java, Python, and Scala in a hybrid role in Frankfurt.
Principal AI Engineer
Join Cere Network as a Principal AI Engineer to drive AI innovation in Web3. Requires 10+ years in AI/ML, NLP, and software development.
Senior Software Engineer (C++, Python & Cloud)
Join VIAVI Solutions as a Senior Software Engineer specializing in C++, Python, and cloud technologies. Work in a hybrid environment in Berlin.
Senior DevOps Engineer - Price & Deal Management
Join Covestro as a Senior DevOps Engineer to drive digital transformation in pricing and deal management with AWS, Docker, and Java expertise.
Software Engineer - Autonomous Driving
Join Applied Intuition as a Software Engineer in Munich to tackle autonomous driving challenges with top ADAS/AV programs.
