Senior ML Engineer (ML/AI)
About The Position
We are looking for an experienced Senior AI/ML Engineer to spearhead the design, development, and deployment of next-generation AI capabilities for our platform. In this role, you won't just train models in a sandbox—you will architect robust, scalable production systems that integrate predictive analytics, Natural Language Processing (NLP), and Generative AI to solve complex, real-world problems. As a senior member of the team, you will bridge the gap between data science research and production-grade software engineering, while mentoring junior engineers and influencing our AI product roadmap. Lyra is for you if you: Thrive on working with brilliant teammates to solve complex, meaningful problems. Are passionate about making a social impact and supporting people at their most challenging moments. Enjoy cross-functional collaboration with physicians, therapists, data scientists, data analysts and product managers.
Requirements
- 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 Haves
- 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 writing production code in Java/Kotlin
- Experience building solutions on cloud infrastructure, particularly AWS
- Experience working with highly sensitive data in a healthcare environment
Responsibilities
- Design, train, fine-tune, and evaluate deep learning, classical ML, and foundational GenAI models (LLMs, diffusion models) to drive core product features.
- Build, scale, and maintain robust ML pipelines for continuous training, evaluation, and real-time/batch inference using modern MLOps frameworks.
- Collaborate with backend and frontend teams to integrate AI models into microservices, ensuring low latency, high availability, and optimal resource utilization.
- Architect scalable data pipelines for preprocessing, vectorizing, and ingestion of massive structured and unstructured datasets.
- Implement rigorous evaluation frameworks for model alignment, bias mitigation, guardrailing, and cost/latency optimization (e.g., quantization, distillation).
- Provide technical leadership, conduct thorough code reviews, and mentor junior/mid-level engineers on best practices in software craftsmanship and ML engineering.
Benefits
- Comprehensive healthcare coverage (including medical, dental, vision, FSA/HSA, life and disability insurances)
- Lyra for Lyrians; coaching and therapy services
- Equity in the company through discretionary restricted stock units
- Competitive time off with pay policies including vacation, sick days, and company holidays
- Paid parental leave
- 401K with up to 3% matching
- Monthly tech allowance
- Well-being perks and activities
- Surprise swag
- Regular community celebration
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
Education Level
No Education Listed
