[Remote] Senior ML Engineer (ML/AI)

Note: The job is a remote job and is open to candidates in USA. Lyra Health is a leading provider of evidence-based mental health care, serving millions globally. They are seeking a Senior ML Engineer to design, develop, and deploy AI capabilities for their platform, integrating predictive analytics and Natural Language Processing to solve complex problems. ResponsibilitiesDesign, train, fine-tune, and evaluate deep learning, classical ML, and foundational GenAI models (LLMs, diffusion models) to drive core product featuresBuild, scale, and maintain robust ML pipelines for continuous training, evaluation, and real-time/batch inference using modern MLOps frameworksCollaborate with backend and frontend teams to integrate AI models into microservices, ensuring low latency, high availability, and optimal resource utilizationArchitect scalable data pipelines for preprocessing, vectorizing, and ingestion of massive structured and unstructured datasetsImplement 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 engineeringSkills 6+ years of experience deploying ML/AI solutions in production environmentsAbility to write high-quality code in PythonExperience building RAG (retrieval-augmented generation) based solutionsExperience setting up and maintaining vector databasesA strong desire to work on ML/AI based productsA desire to learn new technologies quicklyA thoughtful approach to balancing quality and deadlines in fast-paced settingsExcellent communication skills with a talent for building consensus and alignmentStrong organizational skills and the ability to distill complex problems into clear priorities that move the team and business forwardExperience defining and using Protobuf messagesExperience working with Docker and deploying applications to KubernetesExperience with relational and low-latency databasesExperience working with CeleryExperience building RAG (retrieval-augmented generation) based solutionsExperience setting up and maintaining vector databasesExperience writing production code in Java/KotlinExperience building solutions on cloud infrastructure, particularly AWSExperience working with highly sensitive data in a healthcare environmentBenefitsComprehensive healthcare coverage (including medical, dental, vision, FSA/HSA, life and disability insurances)Lyra for Lyrians; coaching and therapy servicesEquity in the company through discretionary restricted stock unitsCompetitive time off with pay policies including vacation, sick days, and company holidaysPaid parental leave401K with up to 3% matchingMonthly tech allowanceWe like to spread joy throughout the year with well-being perks and activities, surprise swag, regular community celebration…and more!Company Overview Lyra helps companies improve access to effective mental health care for their employees. It was founded in 2015, and is headquartered in California City, California, USA, with a workforce of 1001-5000 employees. Its website is http://www.lyrahealth.com. Company H1B SponsorshipLyra Health has a track record of offering H1B sponsorships, with 1 in 2026, 3 in 2025, 4 in 2024, 6 in 2023, 6 in 2022, 1 in 2021, 3 in 2020. Please note that this does not guarantee sponsorship for this specific role.