DESCRIPTION

BASIC QUALIFICATIONS

• Master's degree or above in computer science, machine learning, engineering, or related fields
• 3+ years of programming in Java, C++, Python or related language experience
• 1+ years experience with distributed training frameworks such as VeRL, Megatron, FSDP, DeepSpeed, Ray, or similar systems, and inference engines such as vLLM, TensorRT-LLM, Triton, SGLang, TGI.
• 3+ years’ experience with modeling languages and tools like PyTorch / TensorFlow, R, scikit-learn, numpy, scipy, etc.
• Solid ML background and familiar with NLU, NLG, and LLM training and evaluation.

PREFERRED QUALIFICATIONS

• PhD in computer science, machine learning, engineering, or related fields
• 3+ years experience with distributed training frameworks such as verl, Megatron, FSDP, DeepSpeed, Ray, or similar systems, and inference engines such as vLLM, TensorRT-LLM, Triton, SGLang, TGI.
• Publications at peer-reviewed NLP/ML conferences (e.g. ACL, EMNLP, NAACL, NeurIPS, ICLR, ICML, AAAI, etc.)
• Scientific thinking and the ability to invent, a track record of thought leadership and contributions that have advanced the field.

The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.