Senior ML Engineer (ML/AI)

The role

Design, build and deploy AI and machine learning systems for the Lyra platform, including predictive analytics, NLP and generative AI. Lead technical work and mentor junior engineers while shaping the AI product roadmap.

What you will do

• Design, train, fine-tune and evaluate deep learning, classical ML and GenAI models (LLMs, diffusion models) for core product features
• Build, scale and maintain ML pipelines for continuous training, evaluation and real-time or batch inference
• Integrate AI models into microservices with low latency and high availability
• Architect scalable data pipelines for large structured and unstructured datasets
• Implement evaluation frameworks for model alignment, bias mitigation, guardrailing and cost/latency optimization
• Provide technical leadership, conduct code reviews and mentor junior and mid-level engineers

Required qualifications

• 6+ years of experience deploying ML/AI solutions in production environments
• Ability to write high-quality code in Python
• Experience building RAG (retrieval-augmented generation) based solutions
• Experience setting up and maintaining vector databases
• A strong desire to work on ML/AI based products
• A desire to learn new technologies quickly
• A thoughtful approach to balancing quality and deadlines in fast-paced settings
• Excellent communication skills with a talent for building consensus and alignment
• Strong organizational skills and the ability to distill complex problems into clear priorities that move the team and business forward

Nice to have

• Experience defining and using Protobuf messages
• Experience working with Docker and deploying applications to Kubernetes
• Experience with relational and low-latency databases
• Experience working with Celery
• Experience building RAG (retrieval-augmented generation) based solutions
• Experience setting up and maintaining vector databases
• Experience writing production code in Java/Kotlin
• Experience building solutions on cloud infrastructure, particularly AWS
• Experience working with highly sensitive data in a healthcare environment

About the company

Lyra Health is a provider of evidence-based mental health care, serving over 20 million people through employer and health plan partnerships. Its Lyra Empower platform combines care and AI-powered technology.