Staff AI Engineer

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).