Machine Learning Engineer - LLMs & Generative AI

Job Description

Truveta is seeking a Machine Learning Engineer to contribute to the development of AI systems for healthcare. The ideal candidate will have a strong background in generative AI, large language models (LLMs), and reinforcement learning techniques.

Responsibilities:

• Lead the development, training, and deployment of large language and multimodal foundation models.
• Apply and refine state-of-the-art techniques such as supervised fine-tuning (SFT), reinforcement learning-based methods (e.g., RLHF, RLVR), parameter-efficient fine-tuning (PEFT), prompt tuning, and retrieval-augmented generation (RAG).
• Collaborate with researchers, clinicians, and engineers to design ML-driven solutions.
• Build scalable infrastructure for distributed training of large models.
• Design and evaluate models for robustness, bias mitigation, factual consistency, and explainability.
• Stay current with the latest research in generative AI and contribute back to the community.

Requirements:

• 6+ years of experience in software engineering or machine learning (3+ years with a PhD).
• Experience designing and training LLMs or large-scale generative models.
• Deep expertise in NLP, sequence modeling, and transformer architectures.
• Proficient in Python and ML libraries such as PyTorch or TensorFlow.
• Experience with RL-based fine-tuning and evaluation of generative systems.
• Proven ability to lead technical projects and collaborate across teams.
• Bachelor's degree in Computer Science, Engineering, or a related technical field.

Truveta offers:

• Interesting and meaningful work.
• Comprehensive benefits package.
• Professional development & training opportunities.
• Work/life autonomy via flexible work hours and flexible paid time off.
• Generous parental leave.
• Regular team activities.