Machine Learning Engineer, Gen AI Innovation Center, AWS - Amazon : Job Details

Machine Learning Engineer, Gen AI Innovation Center, AWS

Join to apply for the Machine Learning Engineer, Gen AI Innovation Center, AWS role at Amazon Web Services (AWS).

Role 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 collaborates with clients across industries to deploy customized generative AI applications at scale and works closely with foundational model providers to optimize AI models for Amazon Silicon, enhancing performance and efficiency.

Responsibilities

• Design and implement distributed training pipelines for Large Language Models (LLMs) using tools like Fully Sharded Data Parallel (FSDP) and DeepSpeed.
• Adapt LLMs for new languages, domains, and vision applications through pre-training, fine-tuning, and Reinforcement Learning with Human Feedback (RLHF).
• Optimize AI models for deployment on AWS Inferentia and Trainium using the AWS Neuron SDK and develop custom kernels for performance enhancement.
• Engage with enterprise customers and foundational model providers to understand challenges and co-develop tailored AI solutions.

A Day in the Life

You will develop solutions to improve software architecture, create metrics, automate processes, resolve software defects, and:

• Build impactful Gen AI solutions for a large customer base.
• Participate in design discussions and code reviews.
• Collaborate cross-functionally to drive business solutions.
• Work in a dynamic, startup-like environment focusing on high-priority projects.

About The Team & Culture

At AWS, we value diversity, inclusion, work-life balance, and continuous learning. We encourage candidates of all backgrounds and experience levels to apply. Our culture promotes curiosity, mentorship, and career growth, striving to be Earth's best employer.

Basic Qualifications

• 3+ years of professional software development experience.
• 2+ years of experience in system design or architecture.
• Proficiency in at least one programming language.
• Experience with deep learning/machine learning methods.
• Hands-on experience with generative AI technologies.

Preferred Qualifications

• 3+ years of full-stack development experience.
• Bachelor's degree in computer science or related field.
• Graduate degree (MS or PhD) in relevant fields.
• 1+ years of experience developing or deploying ML models.

We are an equal opportunity employer committed to diversity and inclusion. For accommodations during the application process, visit our website. Compensation varies by location and experience, with additional benefits offered.

#J-18808-Ljbffr