TL;DR

Senior Machine Learning Engineer (AI): Developing and optimizing custom Large Language Models (LLMs) and generative AI solutions with an accent on large-scale distributed training pipelines, model fine-tuning, and optimization for AWS custom AI hardware. Focus on designing scalable training systems, collaborating with enterprise customers, and deploying high-performance AI models on AWS infrastructure.

Location: Remote

Salary: $151,300–$261,500 per year

Company

%hirify_global% is part of Amazon Web Services, a leading global cloud platform innovating in AI and cloud computing technologies.

What you will do

Design and implement distributed training pipelines for large language models using tools like FSDP and DeepSpeed.
Fine-tune and customize LLMs for various languages, domains, and vision applications including RLHF.
Optimize AI models for deployment on AWS Inferentia and Trainium using AWS Neuron SDK and custom kernels.
Collaborate with enterprise customers and foundational model providers to co-develop tailored generative AI solutions.
Drive innovation in end-to-end LLM training pipelines at massive scale.

Requirements

Location: Remote work with preference for US-based candidates.
5+ years of professional software development experience including design and architecture of scalable systems.
Hands-on experience with deep learning, machine learning methods, and generative AI technologies.
Experience mentoring or leading engineering teams.
Bachelor's degree in computer science or equivalent preferred.
Proficiency in at least one ML library or framework and software development lifecycle.

Culture & Benefits

Flexible work-life balance and remote work culture.
Inclusive and diverse team environment with employee-led affinity groups.
Mentorship and continuous career growth opportunities.
Comprehensive compensation including base salary, equity, and benefits.