Machine Learning Engineer - LLMs & Generative AI jobs in United States
cer-icon
Apply on Employer Site
company-logo

Truveta · 1 month ago

Machine Learning Engineer - LLMs & Generative AI

Truveta is the world’s first health provider led data platform with a vision of Saving Lives with Data. We are looking for a Machine Learning Engineer with a deep background in generative AI and large language models to build AI systems for trustworthy healthcare, focusing on developing and deploying large models tailored to clinical and biomedical domains.

AnalyticsData ManagementHealth Care
badNo H1Bnote

Responsibilities

Lead the development, training, and deployment of large language and multimodal foundation models tailored to clinical and biomedical domains
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 cross-functionally with researchers, clinicians, and engineers to design ML-driven solutions that improve healthcare delivery and outcomes
Build scalable infrastructure for distributed training of large models (TPU/GPU clusters, multi-node orchestration)
Design and evaluate models for robustness, bias mitigation, factual consistency, and explainability in healthcare contexts
Stay current with the latest research in generative AI and contribute back to the community through publications and open-source initiatives

Qualification

Generative AILarge Language ModelsReinforcement LearningPythonNLPPyTorchTensorFlowModel InterpretabilityCollaborationProblem-solvingTechnical Leadership

Required

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 (e.g., GPT, PaLM, LLaMA, Claude, Gemma)
Deep expertise in NLP, sequence modeling, and transformer architectures
Proficient in Python and ML libraries such as PyTorch or TensorFlow; strong engineering skills in building scalable ML pipelines
Experience with RL-based fine-tuning (e.g., Reinforcement Learning from Human Feedback) 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

Preferred

PhD or equivalent experience in Machine Learning, NLP, AI, or a related field
Experience in healthcare, biomedical informatics, or clinical data modeling
Familiarity with multi-modal foundation models (e.g., text-image, text-structured data) and cross-modal alignment techniques
Hands-on experience with vector databases, semantic search, and retrieval-based generation
Publications in top-tier ML conferences (NeurIPS, ICML, ACL, EMNLP, ICLR, etc.)
Experience building trustworthy AI systems and applying model interpretability, fairness, and safety frameworks

Benefits

Great benefits package
Comprehensive benefits with strong medical, dental and vision insurance plans
401K plan
Professional development & training opportunities for continuous learning
Work/life autonomy via flexible work hours and flexible paid time off
Generous parental leave
Regular team activities (virtual and in-person as soon as we are able)

Company

Truveta

twittertwittertwitter
company-logo
Truveta is a healthcare data platform that provides EHR data for scientific research.

Funding

Current Stage
Growth Stage
Total Funding
$515M
Key Investors
Microsoft
2025-01-13Series C· $320M
2021-11-09Series Unknown· $100M
2021-09-29Corporate Round

Leadership Team

leader-logo
Terry Myerson
CEO and Co-founder
linkedin
leader-logo
Jay Nanduri
Chief Technical Officer & Co-Founder
linkedin
Company data provided by crunchbase