[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 AI/ML Engineer to design, develop, and deploy AI capabilities for their platform, bridging the gap between data science and software engineering while mentoring junior engineers.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 engineeringSkills6+ 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 environmentCompany OverviewLyra 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.