Remote Position
[Remote] Senior AI/ML Platform Engineer
🌍 Remote · United States
💼 Full-time
Note: The job is a remote job and is open to candidates in USA. Lyra Health provides evidence-based mental health care through technology-enabled solutions. The company is seeking an experienced Senior AI/ML Platform Engineer to build platforms, infrastructure, and services for machine learning and generative AI products. The role involves developing scalable backend infrastructure, building AI services, deploying and managing production applications, and collaborating across teams.
Responsibilities
- Be part of a team working on building our machine learning and generative AI platforms that will be used by teams across the company
- Create the necessary backend infrastructure to enable scalable, resilient and highly performant solutions
- Build services that expose machine learning and AI based products
- Deploy and manage various applications in production
- And of course, you will be coding every day!
Skills
- 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
- Strong organizational skills and the ability to distill complex problems into clear priorities that move the team and business forward
- 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) to track latency, token usage, and output quality
- 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
Company Overview
Company H1B Sponsorship