Senior ML Research Engineer — Multimodal AGI Systems

Amazon; Seattle (WA); On-site; USD 151,300 - 261,500; Full time

Benefits offered by this job

Competitive compensation

Comprehensive medical benefits

Opportunities for professional growth

Job summary

A leading tech company is seeking a passionate Senior Research Engineer specializing in Machine Learning and Artificial General Intelligence. This role involves leading the development of multimodal large-language foundational models. Ideal candidates have over 5 years of software development experience and expertise in machine learning. You'll work with cutting-edge technology and be crucial in advancing the company's AI capabilities. This full-time position is based in Seattle, Washington, and offers competitive compensation and benefits.

Qualifications

• 5+ years of professional software development experience.
• 5+ years of programming with at least one programming language.
• 2+ years of expertise in Machine Learning and/or Model Training.

Responsibilities

• Responsible for pre‑ and post‑training multimodal LLMs.
• Scale training of models on hyper‑large GPU and AWS Trainium clusters.
• Optimize training workflows using distributed training techniques.

Skills

Machine Learning

Software Development

CUDA optimization

Agile Methodologies

CUDA optimization

Education

Master’s degree in machine learning or equivalent

Tools

NeMo

Megatron Core

PyTorch

Jax

Job description

A leading tech company is seeking a passionate Senior Research Engineer specializing in Machine Learning and Artificial General Intelligence. This role involves leading the development of multimodal large-language foundational models. Ideal candidates have over 5 years of software development experience and expertise in machine learning. You'll work with cutting-edge technology and be crucial in advancing the company's AI capabilities. This full-time position is based in Seattle, Washington, and offers competitive compensation and benefits.