Machine Learning Engineer, Generative AI Innovation Center
The Generative AI Innovation Center at AWS empowers customers to harness state‑of‑the‑art AI technologies for transformative business opportunities.
Our multidisciplinary team of strategists, scientists, engineers, and architects collaborates with customers across industries to fine‑tune and deploy customized generative AI applications at scale.
In this role you will drive the development of custom large‑language models (LLMs) across languages, domains, and modalities, fine‑tune state‑of‑the‑art LLMs for diverse use cases, and optimize models for high‑performance deployment on AWS's custom AI accelerators.
Key job responsibilities Large‑Scale Training Pipelines: Design and implement distributed training pipelines for LLMs using tools such as Fully Sharded Data Parallel (FSDP) and Deep Speed, ensuring scalability and efficiency.
LLM Customization & Fine‑Tuning: Adapt LLMs for new languages, domains, and vision applications through continued pre‑training, fine‑tuning, and Reinforcement Learning with Human Feedback (RLHF).
Model Optimization on AWS Silicon: Optimize AI models for deployment on AWS Inferentia and Trainium, leveraging the AWS Neuron SDK and developing custom kernels for enhanced performance.
Customer Collaboration: Interact with enterprise customers and foundational model providers to understand their business and technical challenges and co‑develop tailored generative AI solutions.
Basic Qualifications 3+ years of professional software development experience (non‑internship).
2+ years of industry design or architecture experience for new and existing systems.
Proficiency in at least one software programming language.
Hands‑on experience with deep learning and/or machine learning methods (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+ year of experience deploying or optimizing machine learning models using a recognized ML library or framework.
Equal Opportunity Statement
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
Our inclusive culture empowers Amazonians to deliver the best results for our customers.
If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, please contact our recruiting partner for assistance.
#J-18808-Ljbffr