Machine Learning Infrastructure Engineer - #4694

GRAIL, Inc.
Menlo Park, United States of America
1 month ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Experience level
Senior
Compensation
$ 255K

Job location

Menlo Park, United States of America

Tech stack

Java
Artificial Intelligence
Amazon Web Services (AWS)
Bioinformatics
C Sharp (Programming Language)
C++
Cloud Computing
Computer Programming
Continuous Integration
Distributed Systems
R
Gradle
Python
Machine Learning
Maven
NumPy
TensorFlow
Scientific Computating
Software Engineering
Software Systems
Data Processing
PyTorch
Large Language Models
Spark
Jupyter
Information Technology
Apache Flink
GXP
Docker
Go

Job description

This is a hybrid role based in Menlo Park, CA (moving to Sunnyvale, CA in Fall 2026). Our current hybrid policy requires on-site presence at least 40% of the time, including key in-person collaboration days. At our Menlo Park campus, Tuesdays and Thursdays are the key days where we encourage on-site presence to engage in events and on-site activities. Responsibilities

  • Partner with research teams to identify computational pain points or limitations in performing computational experiments and analyses.
  • Design, build, and evolve software which usefully extends research capabilities, including infrastructure for distributed ML training and evaluation on large controlled genomic datasets.
  • Develop tools and processes that ensure GxP-compliant testing, patchability, and inference reproducibility for classifiers that are promoted to production use.
  • Develop and maintain the research team's software environment, including tools to assess the health, performance, and cost of the system.

These summarize the role's primary responsibilities and are not an exhaustive list. They may change at the company's discretion.

Requirements

The ideal candidate will bring a passion for reliable software infrastructure, distributed computing, reproducible research, and general problem-solving. Due to the highly connected nature of this position, the candidate should be a strong communicator with experience working with multidisciplinary teams., * 5+ years of experience developing software supporting machine learning, scientific computing, or large-scale data processing systems

  • Strong programming skills in Python and a systems-level language such as Golang (preferred), Java, C#, C++, etc.
  • Experience working with modern machine learning frameworks such as PyTorch or TensorFlow
  • Experience with Distributed Computing paradigms (Spark, Ray, Flink, Beam, etc.)
  • A commitment to high-quality professionally engineered software
  • Strong communication skills with the ability to help developers from a wide range of software development backgrounds
  • BS in Computer Science, Engineering, Bioinformatics, or a related field, or equivalent practical experience

Preferred Qualifications

  • Good understanding of container orchestration through Docker and cloud technologies.
  • Experience with scientific computing tools: NumPy, Jupyter, R Notebook, etc.
  • Experience with techniques used in modern AI (including LLM) training
  • Experience with whole genome sequencing, whole exome sequencing, bisulfite sequencing, and/or whole transcriptome sequencing data
  • Practical experience setting up continuous integration systems, along with expertise in at least one build tool (e.g. Bazel (preferred), Buck, Maven, Gradle)
  • Familiarity with AWS services, best practices, and security
  • Advanced degree (MS or PhD) in computer science, engineering, bioinformatics or a related discipline

Benefits & conditions

The expected, full-time, annual base pay scale for this position is $190k-$255k.

This role may be eligible for other forms of compensation, including an annual bonus and/or incentives, subject to the terms of the applicable plans and Company discretion. This range reflects a good-faith estimate of the range that the Company reasonably expects to pay for the position upon hire; the actual compensation offered may vary depending on factors such as the candidate's qualifications. Employees in this role are also eligible for GRAIL's comprehensive and competitive benefits package, offered in accordance with our applicable plans and policies. This package currently includes flexible time-off or vacation; a 401(k) retirement plan with employer match; medical, dental, and vision coverage; and carefully selected mindfulness programs.

About the company

Our mission is to detect cancer early, when it can be cured. We are working to change the trajectory of cancer mortality and bring stakeholders together to adopt innovative, safe, and effective technologies that can transform cancer care. We are a healthcare company, pioneering new technologies to advance early cancer detection. We have built a multi-disciplinary organization of scientists, engineers, and physicians and we are using the power of next-generation sequencing (NGS), population-scale clinical studies, and state-of-the-art computer science and data science to overcome one of medicine's greatest challenges. GRAIL is headquartered in the bay area of California, with locations in Washington, D.C., North Carolina, and the United Kingdom. It is supported by leading global investors and pharmaceutical, technology, and healthcare companies., GRAIL is a healthcare company whose mission is to detect cancer early, when it can be cured. GRAIL is focused on alleviating the global burden of cancer by developing pioneering technology to detect and identify multiple deadly cancer types early. The company is using the power of next-generation sequencing, population-scale clinical studies, and state-of-the-art computer science and data science to enhance the scientific understanding of cancer biology, and to develop its multi-cancer early detection blood test. GRAIL is headquartered in Menlo Park, CA with locations in Washington, D.C., North Carolina, and the United Kingdom. It is supported by leading global investors and pharmaceutical, technology, and healthcare companies. For more information, please visit www.grail.com.

Apply for this position