Amazon Jobs
Machine Learning Engineer, Model Customization, Generative AI Innovation Center
Amazon Jobs, Mountain View, California, us, 94039
Overview
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 and optimize models for Amazon Silicon.
Key Responsibilities
Design and implement distributed training pipelines for LLMs using FSDP, DeepSpeed, ensuring scalability and efficiency.
Adapt LLMs for new languages, domains, and vision applications through continued pre‑training, fine‑tuning, and RLHF.
Optimize AI models for deployment on AWS Inferentia and Trainium, developing custom kernels and leveraging the Neuron SDK.
Interact with enterprise customers and foundational model providers, co‑de‑veloping tailored generative AI solutions to solve their business challenges.
Qualifications
3+ years of non‑internship professional software development experience.
2+ years of experience designing or architecting scalable systems.
Bachelor’s degree in computer science or equivalent.
Hands‑on experience with deep learning, machine learning methods, and generative AI technology.
Full software development life cycle experience including coding standards, code reviews, source control, build, testing, and operations.
Experience developing, deploying, or optimizing ML models using recognized libraries or frameworks.
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
#J-18808-Ljbffr
Key Responsibilities
Design and implement distributed training pipelines for LLMs using FSDP, DeepSpeed, ensuring scalability and efficiency.
Adapt LLMs for new languages, domains, and vision applications through continued pre‑training, fine‑tuning, and RLHF.
Optimize AI models for deployment on AWS Inferentia and Trainium, developing custom kernels and leveraging the Neuron SDK.
Interact with enterprise customers and foundational model providers, co‑de‑veloping tailored generative AI solutions to solve their business challenges.
Qualifications
3+ years of non‑internship professional software development experience.
2+ years of experience designing or architecting scalable systems.
Bachelor’s degree in computer science or equivalent.
Hands‑on experience with deep learning, machine learning methods, and generative AI technology.
Full software development life cycle experience including coding standards, code reviews, source control, build, testing, and operations.
Experience developing, deploying, or optimizing ML models using recognized libraries or frameworks.
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
#J-18808-Ljbffr