Machine Learning Engineer, Model Customization (Generative AI)

TL;DR

Machine Learning Engineer (Generative AI/LLM): Building and optimizing custom Large Language Models across languages, domains, and modalities with an accent on distributed training pipelines, fine-tuning, and model optimization for AWS AI accelerators. Focus on tackling end-to-end LLM training pipelines at massive scale and delivering next-generation AI solutions for top AWS clients.

Location: Must be based in the United States. Applicants from Los Angeles County are also considered.

Salary: $129,300–$223,600/year

Company

hirify.global (AWS) is a comprehensive cloud platform that empowers customers to leverage state-of-the-art AI technologies for transformative business opportunities.

What you will do

• Design and implement distributed training pipelines for LLMs, ensuring scalability and efficiency.
• Adapt LLMs for new languages, domains, and vision applications through continued pre-training, fine-tuning, and Reinforcement Learning with Human Feedback (RLHF).
• Optimize AI models for deployment on AWS Inferentia and Trainium using the AWS Neuron SDK.
• Collaborate with enterprise customers and foundational model providers to co-develop tailored generative AI solutions.

Requirements

• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship design or architecture experience in new and existing systems.
• Bachelor's degree in computer science or equivalent.
• Hands-on experience with deep learning and/or machine learning methods (e.g., for training, fine-tuning, and inference).
• Hands-on experience with generative AI technology.

Culture & Benefits

• Emphasis on work-life harmony and flexibility in the working culture.
• Inclusive team culture promoted by employee-led affinity groups and inclusion events.
• Access to knowledge-sharing, mentorship, and career-advancing resources.
• Opportunity to innovate at the forefront of AI within a diverse team of technical experts.