Sr. Machine Learning Engineer, Amazon FAI

Job description

As an Sr. MLE on our team, you will design, build, and maintain systems for building frontier quality assets to train and evaluate our GenAI models. You’ll also work closely with our Applied Scientists to develop tools that support our modeling and evaluation team.

We operate at Amazon's large scale with the energy of a nimble start-up. If you have a learner's mindset, enjoy solving challenging problems, and value an inclusive team culture, you will thrive in this role, and we hope to hear from you.

Key job responsibilities

• Develop and maintain key tooling for building agentic pipelines to create frontier-level assets, such as RL gyms.
• Work closely with product teams, and scientists to create new techniques and tooling around emerging research capabilities and evaluating models to meet customer needs.
• Manage project prioritization, deliverables, timelines, and stakeholder communication.
• Illuminate trade-offs, educate the team on best practices, and influence technical strategy.
• Collaborate closely with Applied Scientists to investigate and plan experiments, prototype new technology, and evaluate technical feasibility.
• Operate in a dynamic environment to deliver high-quality software against tight schedules.

About the team

The FAI team has the mission to push the envelope in GenAI with LLMs, in order to provide the best-possible experience for our customers.