Machine Learning Engineer III (ML Core)

Who We Are

PathAI's mission is to improve patient outcomes with AI-powered pathology. Our platform promises substantial improvements to the accuracy of diagnosis and the efficacy of treatment of diseases like cancer, leveraging modern approaches in machine learning and artificial intelligence. We have a track record of success in deploying AI algorithms for histopathology in translational research, pathology labs and clinical trials.  Rigorous science and careful analysis is critical to the success of everything we do. Our team, composed of diverse employees with a wide range of backgrounds and experiences, is passionate about solving challenging problems and making a huge impact on patient outcomes.

Where You Fit 

We’re seeking a Machine Learning Engineer with experience in ML Ops and tooling who also has a strong understanding of applied machine learning. You’ll work at the intersection of ML engineering and operations, building robust tools, systems, and platforms that empower our foundational models and accelerate the end-to-end ML lifecycle.

What You’ll Do

• Design, build, and optimize ML infrastructure and platforms supporting foundational base models, enabling rapid iteration from training to deployment.
• Develop robust and scalable tooling to streamline end-to-end ML workflows including data ingestion, dataset creation & management, automated training, comprehensive evaluation, and seamless model deployment.
• Collaborate closely with ML engineers, researchers, software engineers, and product managers to effectively integrate and deploy innovative ML methods and base models into production systems.
• Implement and optimize automated ML operations through advanced CI/CD pipelines, leveraging technologies like Kubernetes, Docker, Helm, and Airflow for orchestration and infrastructure-as-code.
• Standardize and enhance model performance, reliability, data efficiency, and inference latency across diverse ML use-cases, utilizing rigorous benchmarking and iterative improvement methodologies.
• Engage in technical research and prototyping to evaluate new tools, frameworks, and methodologies that enhance MLOps and applied ML capabilities.
• Champion best practices, contribute to documentation, mentor team members, and participate actively in knowledge sharing and continuous improvement initiatives.
• You will have the opportunity to work in a company where all employees put patients first. We believe that every team member provides valuable contributions to our success, and no task is too small for anyone if it's important to our company goals. Every PathAI employee is a contributor to our mission to pioneer better patient care by providing the best, most innovative AI tools to biotech, pathologists, clinicians and healthcare organizations. You will work alongside and with leading innovators in the field of AI and medicine and you will play a critical role in product development to impact patient outcomes.

What You Bring

• Bachelor’s or Master’s degree plus 3-5 years of professional experience with a focus on artificial intelligence and machine learning.
• Proven track record in developing, deploying, and maintaining state-of-the-art ML models in scalable production environments.
• Strong experience in building and managing large-scale ML pipelines, feature and embedding stores, model monitoring, and performance evaluation systems.
• Hands-on proficiency with MLOps technologies such as Kubernetes, AWS, Docker, Terraform, Helm, Kafka, Airflow, Kubeflow, Argo, Ray, and GitLab/GitHub Actions for end-to-end ML lifecycle management.
• Excellent proficiency in Deep Learning frameworks (Pytorch/Tensorflow), Python (including Scipy, Numpy, Pandas) and software engineering skills. Strong analytical and quantitative skills.
• Knowledge of recent advances in machine learning and computer vision concepts, including but not limited to transformer-based models, self-supervised learning, advanced segmentation models, inference-optimization techniques.
• Strong communication skills and the ability to collaborate effectively in a cross-functional environment.
• Intellectual curiosity and the ability to learn quickly in a complex space.
• Publications or research in fields related to machine learning and biomedical science are a bonus.
• Proficiency in using AI Assistants for code development (Cursor, CoPilot) is a bonus.

We Want To Hear From You

At PathAI, we are looking for individuals who are team players, are willing to do the work no matter how big or small it may be, and who are passionate about everything they do. If this sounds like you, even if you may not match the job description to a tee, we encourage you to apply. You could be exactly who we're looking for. 

PathAI is an equal opportunity employer, dedicated to creating a workplace that is free of harassment and discrimination. We base our employment decisions on business needs, job requirements, and qualifications — that's all. We do not discriminate based on race, gender, religion, health, personal beliefs, age, family or parental status, or any other status. We don't tolerate any kind of discrimination or bias, and we are looking for teammates who feel the same way.