Staff Machine Learning Engineer

Summary

Join Atropos Health, an award-winning AI company, as a Staff Machine Learning Engineer to contribute to the development and improvement of AI-powered healthcare products, such as ChatRWD and Alexandria. You will design and implement scalable AI solutions, collaborate with cross-functional teams, establish best practices for LLM application, and ensure solutions align with system vision and scalability goals. This role requires expertise in AI/ML technologies, especially LLMs, and strong experience with Python, cloud platforms, and distributed systems. You will also interact with senior business leaders and contribute to the company's roadmap. The ideal candidate will have a strong background in software architecture, a deep understanding of modern software engineering principles, and experience in healthcare technology or regulated industries.

Requirements

• Bachelor’s degree in Computer Science, Engineering, or a related field (or equivalent experience)
• Expertise in AI/ML technologies, especially Large Language Models (LLMs), autonomous agents, and modern machine learning frameworks
• Strong hands-on experience with Python, cloud platforms (AWS preferred), and distributed systems architecture
• 10+ years of software/ML engineering experience
• 5+ years experience focused on AI/ML application development
• 2+ years of experience designing, implementing, and optimizing enterprise-scale applications, with a focus on workflow automation through AI
• Proven experience as a software architect or senior developer, with a track record of leading complex technical initiatives
• Deep knowledge of modern software engineering principles, architectural best practices, and emerging AI trends
• Excellent written and verbal communication skills, with the ability to convey complex technical concepts to diverse audiences
• Demonstrated ability to thrive in fast-paced, high-growth, and rapidly evolving environments
• Strong analytical and problem-solving skills, with a high level of judgment and creativity in designing innovative solutions
• Understanding of ethical AI principles, governance frameworks, and regulatory compliance (HIPAA, GDPR, SOC2)
• Ability to work effectively in a remote-first environment, ensuring high-quality virtual interactions with minimal distractions
• Willingness to travel periodically for team collaboration and company events

Responsibilities

• Design and guide the architecture and participate in the implementation of scalable, AI-powered solutions to fit into or replace workflow
• Collaborate with engineering, product, and leadership teams to define and deliver new features and enhancements
• Establish best practices and architectural patterns for the application of LLMs in the process of generating clinical evidence from internal, structured data and unstructured sources
• Set up and provide technical oversight for evaluation mechanisms to ensure solutions align with the system vision, scalability goals, and best practices
• Interact with senior business leaders for all major capability enhancements (roadmap) and provide/own the proposed architectural solution. Participate in road mapping activities with all stakeholders
• Contribute to day-to-day technical deliverables
• Stay at the forefront by continuously exploring emerging technologies to keep our business competitive and innovative

Preferred Qualifications

• A Master’s degree in Computer Science or related discipline is preferred
• Experience in healthcare technology or regulated industries is preferred

Benefits

• Health Benefits . Our benefit package includes employer paid Medical, Dental, Vision, Life, STD, and LTD insurance. We also offer FSA
• Parental Leave . We offer up to twelve weeks of paid leave for new parents who have been at the company for 6+ months
• Financial Wellness . Save for retirement through our 401k plan with Human Interest
• Flexible Work Environment . We're a remote first company with a flexible vacation policy
• Home Office Stipend . All full-time employees will receive a stipend to support their remote office setup
• Offsites . As a remote company we take time 2-3 times a year to get together in small teams and all together as a company