Senior Machine Learning Engineer, Generative AI, AGI Inference Engine

A global technology company leading in e-commerce, cloud computing, AI, and digital streaming.
Warsaw, PolandGdańsk, PolandKraków, Poland
$150,000 - $250,000
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Senior Machine Learning Engineer, Generative AI, AGI Inference Engine

The Artificial General Intelligence (AGI) Inference Engine team at Amazon is seeking a Senior ML Engineer to spearhead research and implementation of cutting-edge techniques in Generative AI model inference. This role sits at the intersection of advanced machine learning and practical engineering, focusing on optimizing large language models and diffusion models for production environments.

The position offers unique opportunities to work with state-of-the-art AI technologies, including LLMs and diffusion models, while utilizing Amazon's best-in-class GPU and AWS Neuron hardware. As a senior engineer, you'll have the freedom to explore new ideas, conduct research, and directly influence Amazon's AGI roadmap through your recommendations and implementations.

The team's mission is critical to Amazon's AI infrastructure, building fast, accurate, and cost-efficient large language and diffusion model inference solutions that power various Amazon businesses. You'll be working in an environment that values independence and creativity, where you can read recent papers, explore ideas, and freely collaborate with others.

This role is perfect for someone who combines deep technical expertise in machine learning with practical software engineering skills. You'll be expected to mentor junior engineers, lead by example in following best practices, and work cross-functionally to solve complex problems in the GenAI space.

The position offers the opportunity to work with one of the world's leading technology companies, access to cutting-edge AI infrastructure, and the chance to make significant contributions to the future of AI technology. Join us in pushing the boundaries of what's possible in artificial general intelligence and help shape the future of how Amazon delivers value to millions of customers worldwide.

Last updated 2 days ago

Responsibilities For Senior Machine Learning Engineer, Generative AI, AGI Inference Engine

  • Design, develop, test and deploy inference solutions for high-end LLMs and diffusion models
  • Explore emerging inference optimization techniques for LLMs and diffusion models
  • Design and conduct experiments, analyze results and influence Amazon AGI roadmap
  • Optimize performance of LLMs and diffusion models on GPU and AWS Neuron hardware
  • Collaborate with cross-functional teams of engineers and scientists
  • Mentor and guide junior engineers
  • Lead operational efforts and ensure team follows best practices

Requirements For Senior Machine Learning Engineer, Generative AI, AGI Inference Engine

Python
  • Bachelor's degree in computer science or equivalent
  • Several years of non-internship professional software development experience
  • Several years of programming experience
  • Experience with design patterns, reliability and scaling of systems
  • Knowledge of Machine Learning algorithms and techniques
  • Experience as a thesis supervisor, mentor, tech lead or leading an engineering team
  • Experience with Python, PyTorch, and C++ programming and performance optimization
  • Knowledge of LLMs and/or diffusion models
  • Strong communication and collaboration skills

Interested in this job?

Jobs Related To Amazon Senior Machine Learning Engineer, Generative AI, AGI Inference Engine

Senior Software Development Engineer - Amazon Music Machine Learning

Senior ML Engineer role at Amazon Music, leading recommendation systems development and team mentoring in Berlin.

Emulation Engineer, AWS Annapurna Labs

Senior Emulation Engineer position at Amazon working on AWS Inferentia and ML acceleration hardware, offering competitive compensation and benefits.

Machine Learning Engineer, Generative AI Innovation Center

Senior Machine Learning Engineer role at Amazon's GenAI Innovation Center, focusing on developing advanced ML solutions and generative AI applications for enterprise customers.

Senior Software Development Engineer, Enterprise Personalization

Lead AI-powered personalization initiatives at Amazon, developing sophisticated systems that revolutionize customer experiences across all Amazon businesses.

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Senior ML Engineer role at AWS focusing on distributed training systems and ML accelerators, offering competitive compensation and growth opportunities.