Senior AI/ML Platform Engineer

The role

You will build the machine learning and generative AI platforms that teams across the company use. You will write backend infrastructure and services that expose ML and AI products, deploy and manage applications in production, and code every day.

What you will do

• Build the machine learning and generative AI platforms used by teams across the company
• Create scalable, resilient and highly performant backend infrastructure
• Build services that expose machine learning and AI based products
• Deploy and manage applications in production
• Code every day
• Take ownership of your work and lead cross-functional projects
• Provide guidance and mentorship to other team members

Required qualifications

• 6+ years of experience delivering production ready products
• Ability to write high-quality code in Python
• Experience with modern async Python web frameworks (e.g., FastAPI) and handling data streaming (WebSockets, Server-Sent Events)
• Experience working with Docker and deploying applications to Kubernetes
• Experience with relational and low-latency databases
• Experience building RAG based AI solutions
• Experience setting up and maintaining vector databases
• Familiarity with LLM orchestration frameworks (e.g., LangChain, LlamaIndex) and integrating with foundation model APIs (e.g., OpenAI, Anthropic, AWS Bedrock)
• A desire to learn new technologies quickly
• A love of building systems from scratch
• A thoughtful approach to balancing quality and deadlines in fast-paced settings
• Excellent communication skills with a talent for building consensus and alignment

Nice to have

• Experience with Graph Databases (e.g., Neo4j, AWS Neptune) and integrating them into AI/Knowledge systems (e.g., GraphRAG)
• Experience with LLM observability and monitoring tools (e.g., Langfuse)
• Experience working with Celery
• Experience defining and using Protobuf messages
• 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

Before you apply

• Lyra is an Equal Opportunity Employer and does not discriminate on protected categories

About the company

Lyra Health is a provider of evidence-based mental health care, serving more than 20 million people globally in partnership with employers. Its AI-powered platform, Lyra Empower, combines care and technology solutions.