Machine Learning Engineer, Generative AI Innovation Center
Join AWS as a Machine Learning Engineer in our Generative AI Innovation Center, where multidisciplinary teams of strategists, scientists, engineers, and architects collaborate with customers across industries to fine‑tune and deploy customized generative AI applications at scale.
Working closely with foundational model providers, you will design, build, and optimize large language models (LLMs) for diverse use cases, ensuring high‑performance deployment on AWS's custom AI accelerators.
Responsibilities Design and implement distributed training pipelines for LLMs using tools such as Fully Sharded Data Parallel (FSDP) and Deep Speed, ensuring scalability and efficiency.
Adapt LLMs for new languages, domains, and vision applications through continued pre‑training, fine‑tuning, and Reinforcement Learning with Human Feedback (RLHF).
Optimize AI models for deployment on AWS Inferentia and Trainium, leveraging the AWS Neuron SDK and developing custom kernels for enhanced performance.
Collaborate with enterprise customers and foundational model providers to understand business and technical challenges, co‑developing tailored generative AI solutions.
Day in the Life
AWS Global Services brings together experts from across AWS to help customers design, build, operate, and secure their cloud environments.
You'll be part of a diverse team of technical experts in dozens of countries, helping customers achieve more with the AWS cloud.
About the Team
AWS values diverse experiences.
Even if you do not meet all the qualifications and skills listed in the job description, we encourage candidates to apply.
If your career is just starting, hasn't followed a traditional path, or includes alternative experiences, don't let it stop you from applying.
Qualifications Basic Qualifications 3+ years of non‑internship professional software development experience.
2+ years of experience designing or architecting new and existing systems.
Experience programming with at least one software programming language.
Hands‑on experience with deep learning and/or machine learning methods (e.g., training, fine‑tuning, inference).Hands‑on experience with generative AI technology.
Preferred Qualifications 2+ years of full software development life cycle experience, including coding standards, code reviews, source control, build processes, testing, and operations.
Bachelor's degree in computer science or equivalent.
1+ years of experience deploying or optimizing machine learning models using a recognized ML library or framework.
Acknowledgement Of Country
In the spirit of reconciliation, Amazon acknowledges the Traditional Custodians of country throughout Australia and their connections to land, sea, and community.
We pay our respect to their elders past and present and extend that respect to all Aboriginal and Torres Strait Islander peoples today.
Equal Opportunity Statement
Amazon is an equal opportunity employer—no discrimination on the basis of veteran status, disability, or other legally protected status.
If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, please visit Amazon accommodations for more information.
Company: Amazon Web Services Australia Pty Ltd #J-18808-Ljbffr