Forward Deployed Engineer (Model Customization and Optimization), AWS Forward Deployed Engineering

What you'll do

• Translate customer goals into model evaluation criteria and technical plans; deliver end-to-end model workstreams.
• Build and optimize model post-training, distributed training, and production inference systems, including workloads on AWS Trainium.
• Create reusable tools and patterns and share customer learnings with AWS service teams.

What they ask for

• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship experience designing or architecting new and existing systems, including design patterns, reliability, and scaling.
• Experience programming with at least one software programming language.
• Hands-on experience with generative AI technology, such as LLMs, vision-language models, or speech models.

Nice to have

• Bachelor's degree in computer science or equivalent.
• Experience in large-scale distributed LLM training, LLM post-training, model optimization, or deploying and operating ML models in customer production environments.

Pay

165,200.00 - 223,600.00 USD annually (base salary); 143,700.00 - 194,400.00 USD annually (base salary); 158,100.00 - 213,800.00 USD annually (base salary); 143,700.00 - 194,400.00 USD annually (base salary); 143,700.00 - 194,400.00 USD annually (base salary)

Tools

PyTorch FSDP2, Megatron-LM, SGLang, vLLM, AWS Trainium, AWS Neuron SDK, Amazon Bedrock, Amazon SageMaker AI

AI focus

Customize and post-train generative AI models; build evaluation and training pipelines; optimize model training and inference; deploy models as production services.