Staff AI Engineer
IMO Health
Full-time
Remote
United States
Artificial Intelligence
At IMO Health, we combine strengths in software development, artificial intelligence, and clinical expertise to create AI-driven solutions that enhance access to reliable health information, support clinical decision-making, and improve patient outcomes.
We are seeking a Staff AI Engineer to join our Software Engineering team, with strong background in designing, building, deploying, and maintaining AI solutions in production, with an emphasis on software engineering principles. This role is critical in bridging the gap between the theoretical capabilities of AI and its practical implementation by operationalizing AI models, building AI systems that are scalable, robust, and reliable in real-world environments.
The ideal candidate is a hands-on engineer with strong software engineering, data engineering, MLOps, DevOps, and cloud engineering expertise, capable of owning the full development lifecycle – from designing, architecting, and developing AI solutions to deployment, monitoring, and optimization. You will work closely with cross-functional partners who research and prototype models and integrate AI features into IMO Health products.
WHAT YOU’LL DO:
- Collaborate with cross-functional teams to transition AI/ML models from prototypes into scalable, production-ready systems.
- Lead system design and architecture discussions, bringing expertise in AI engineering, MLOps, and AI deployment best practices.
- Write high-quality source code, including detailed documentation and high-level technical designs in collaboration with other engineers.
- Develop and maintain AI-driven applications, ensuring high performance, scalability, reliability, and security, optimizing for latency, throughput, and cost.
- Integrate large language models (LLMs), generative AI, and NLP solutions into IMO Health’s products, with a focus on unstructured clinical data.
- Apply containerization (Docker, Kubernetes) and Infrastructure-as-Code to manage production environments.
- Collaborate in an Agile environment with cross-functional teams, aligning technical solutions with product and business goals.
- Implement creative solutions to technical challenges, applying out-of-the-box thinking to improve existing products.
- Conduct and lead root cause analysis on defects, ensuring fixes are implemented correctly.
- Guide Agile teams on best practices across all SDLC activities. Mentor junior developers.
- Champion technical standards and architectural alignment.
- Advocate for defining and testing Non-Functional Requirements (Availability, Scalability, Reliability, and Maintainability).
- Collaborate with team members while fostering trust, commitment, and accountability.
- Drive discovery and minimization of technical debt, proactively addressing implications and consequences of technical decisions.
WHAT YOU’LL NEED:
- 8+ years of professional experience in software engineering, AI/ML engineering, or related roles, building production-grade web applications.
- Bachelor’s or Master’s degree in Computer Science, Engineering, or a related technical field (or equivalent experience).
- Strong foundation in computer science fundamentals, including data structures, algorithms, design patterns, operating systems, and networking.
- Strong coding skills in Python or Java, with experience in software engineering best practices.
- Experience in developing AI-powered applications, including LLM integration, prompt engineering, and agentic concepts.
- Experience fine-tuning and deploying large language models (LLMs) and generative AI solutions in production.
- Solid hands-on experience with cloud platforms (AWS or Azure), containerization, and Infrastructure-as-Code.
- Experience with MLOps tools and workflows (e.g., MLflow, SageMaker, Kubeflow).
- Experience with CI/CD pipelines, automation, and monitoring.
- Working knowledge of NLP concepts (tokenization, embeddings, classification, sequence modeling) — healthcare domain exposure is a plus.
- Experience working with Elasticsearch and vector databases for embedding-based search and retrieval.
- Ability to translate business needs into optimal technical implementations while balancing technical debt and delivery speed.
- Strong problem-solving mindset to handle complex technical challenges and make informed architectural decisions.
- Experience in high-performance, scalable system design.
- Excellent collaboration skills, working effectively with cross-functional teams including Product, Architecture, and Data Science.
- Passion for AI technologies and innovation, particularly in applications for healthcare.
- Excellent communication and collaboration skills in cross-functional, distributed teams.
- Self-starter with the ability to work independently and contribute from day one.
NICE TO HAVE:
- Experience with clinical or healthcare AI applications.
- Familiarity with Hugging Face, PyTorch, TensorFlow, or other modern ML frameworks.
- AWS Associate-level certification (Machine Learning Engineer or Solutions Architect).
$170,000 - $250,000 a year
Compensation at IMO Health is determined by job level, role requirements, and each candidate’s experience, skills, and location. The listed base pay represents the target for new hires with individual compensation varying accordingly. These figures exclude potential bonuses, equity, or sales incentives, which may also be part of the total compensation package. Our recruiter will provide additional details during the hiring process. IMO Health also offers a comprehensive benefits package. To learn more, please visit IMO Health’s Careers Page.
We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.