GRAIL is seeking a Staff Machine Learning Infrastructure Engineer for the Research Platform Engineering team. This is a software engineering role, charged with building and supporting systems executing machine learning and other analysis workflows on controlled data. You will empower computational biologists, data scientists, and statisticians in their quest to develop and refine powerful diagnostic products, by enabling efficient and flexible exploratory research and classifier development, and smoothing the productionization of their work.
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.
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.
Required Qualifications
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
The expected, full-time, annual base pay scale for this position is $190k-$255k.
What the Team is Saying
.JPG)
.JPG)
GRAIL Compensation & Benefits Highlights
-
Healthcare Strength — Comprehensive medical, dental, and vision coverage is paired with mental‑health resources and disability insurance, with some roles noting strong employer support for HSAs on high‑deductible plans.
-
Equity Value & Accessibility — New‑hire equity grants with clear vesting schedules and an active ESPP make ownership a meaningful part of total rewards, with equity described as straightforward and accessible.
-
Leave & Time Off Breadth — Flexible time off policies, paid holidays, and routine company breaks (e.g., summer and winter) are highlighted across locations, with additional generous vacation allowances in the UK.
GRAIL Insights
Similar Jobs
.png)
What We Do
GRAIL is a healthcare company whose mission is to detect cancer early, when it can be cured. GRAIL is using the power of high-intensity 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 and commercialize pioneering products.
Why Work With Us
Everything we do is guided by our mission to detect cancer early, when it can be cured. It’s the reason we’re here, and it’s no small task. The right people make all the difference. That’s why we’re looking for those who strive to share their knowledge, contribute their skills, inspire each other and commit to something bigger than themselves.
Gallery
GRAIL Offices
Hybrid Workspace
Employees engage in a combination of remote and on-site work.
GRAIL has a variety of work types depending on the roles. Some roles are onsite like a lab role, some are fully remote like our Galleri Sales Consultant roles. Others are hybrid with 2-3 days onsite. Typically Tuesday and Thursday.
