Machine Learning Engineer, Generative AI Innovation Center

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 DeepSpeed, 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 ...