Back to Job Listings

Senior Machine Learning Engineer (Performance)

Boston, MA or Remote

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. Rigorous science and careful analysis is critical to the success of everything we aspire to do. Our team, comprising of diverse employees with a wide range of backgrounds and experiences, is passionate about solving challenging problems and making a huge impact.

We're looking for experienced machine learning scientists to join our team and help solve some of these challenges.  You will work with the MLops team to identify performance bottlenecks in existing ML inference and training workflows and help to optimize them.

What you'll do:

  • Work closely with teams of machine learning engineers to develop and optimize machine learning models with respect to functionality, performance, reliability, and scalability.
  • Research and develop new approaches to model inference that allow PathAI to efficiently and rapidly interpret gigapixel images.
  • Collaborate with a cross-functional team of engineers, data scientists and product managers to clarify objectives and expectations, identify obstacles to success and generate documentation to reflect technical decisions and project plans.
  • Developing libraries and maturing our platform for accelerating our ability to develop, evaluate and monitor machine learning models.

You will have the opportunity to work in a company culture that puts employees first, is relaxed but focused and hard-working. Every PathAI employee is a contributor to our mission to pioneer better patient care by providing the best, most innovative deep learning tools to biotech, pathologists, clinicians and healthcare organizations. You will work alongside and with leading innovators in the field of deep learning AI and medicine and you will play a critical role in algorithm development.

What you'll need:

To be successful in this role with us, you'll at least need:

  • Masters or PhD in Computer Science, Engineering, Data Science or related field required, preferably with a focus on artificial intelligence or equivalent industry experience
  • At least four years plus of relevant work experience
  • Artificial intelligence software development experience using Python and C 
  • Publications or research in fields related to artificial intelligence
  • At least one years of experience working with low-level profiling & debugging tools 
  • At least one year of experience wrapping low-level C code to be used in Python and balancing extensibility, maintainability, and performance 
  • At least 2 years of experience writing production-grade C or C++ for HPC applications
  • Performance engineering experience in Vectorization, Multi-threading (race-conditions, lock-free algorithms) and Multiprocessing - Interprocess communication (queues, shared memory, process forking)
  • Intellectual curiosity and the ability to learn quickly in a complex space 

Preferred Qualifications:

  • Proven experience consulting with engineers / researchers
  • Experience in Fortran or Rust for HPC applications 
  • Experience in production Python development 
  • Proven experience consulting with engineers/researchers who only have Python experience and teaching techniques to optimize their code directly in Python or using Python wrapped lower-level languages
  • Awareness of the latest computer architecture trends for HPC
  • Knowledge of Docker / Kubernetes  
  • Advanced performance engineering experience: MPI / OpenMP multiprocessing, spatial/temporal locality, GPU basics (multiple ALUs, VRAM, CUDA experience), SIMD/MIMD distinction 


For the right candidate, we'll offer a competitive salary plus equity. Your compensation is rounded out by a strong benefits package:

  • Flexible work hours, with work-from-home options available 
  • Three weeks of paid leave per year, an additional two weeks of sick time, plus extended holidays and team-approved leave
  • Ten days of 100% subsidized childcare per year
  • Healthcare, vision, and dental insurance plans (HMO or PPO), with voluntary add-ons available for dependent care, life, and accident coverage
  • Commuter benefit available for public transit or parking

Most importantly, you'll be doing important work with a team of people you'll genuinely enjoy spending the day with. 

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.

PathAI does not accept unsolicited submissions from third-parties. 

With the rise of employment phishing scams, especially during the current COVID-19 situation, we ask candidates to be extra vigilant. Be careful about providing any kind of personal information, such as location, age, or financial information to recruiters who may be impersonating PathAI employees. PathAI does not require payment for interviews or for applicants to purchase their own work supplies for reimbursement.