Job Title

Machine Learning Engineer, Generative AI Innovation Center The Generative AI Innovation Center at AWS empowers customers to harness stateoftheart AI technologies for transformative business opportunities. Our multidisciplinary team of strategists, scientists, engineers, and architects collaborates with customers across industries to finetune and deploy customized generative AI applications at scale. In this role you will drive the development of custom largelanguage models (LLMs) across languages, domains, and modalities, finetune stateoftheart LLMs for diverse use cases, and optimize models for highperformance deployment on AWSs custom AI accelerators. Key job responsibilities LargeScale Training Pipelines: Design and implement distributed training pipelines for LLMs using tools such as Fully Sharded Data Parallel (FSDP) and DeepSpeed, ensuring scalability and efficiency. LLM Customization & FineTuning: Adapt LLMs for new languages, domains, and vision applications through continued pretraining, finetuning, 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 codevelop tailored generative AI solutions. Basic Qualifications 3+ years of professional software development experience (noninternship). 2+ years of industry design or architecture experience for new and existing systems. Proficiency in at least one software programming language. Handson experience with deep learning and/or machine learning methods (training, finetuning, inference). Handson 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. Bachelors 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