Key Facts

Remote From:

Full time

English

Hard Skills

Nvidia CUDA JAX-WS Computer Clusters Continuous Delivery TensorFlow Model-Based Design Quantization PyTorch (Machine Learning Library) Property Ownership Functional Excellence Iterative Method

Other Skills

•
Communication
•
Teamwork

Roles & Responsibilities

Extensive experience training and deploying large-scale models (7B+ parameters) with deep expertise in PyTorch, JAX, or TensorFlow.
Strong background in low-level optimization including quantization, CUDA/Triton, and hardware-specific tuning for GPU/TPU/HPU clusters.
Proven track record of handling massive datasets (2T+ tokens) and building scalable data processing workflows for high-throughput training.
Experience designing and implementing scalable training/inference infrastructure for massive multi-omic datasets and large-scale biological data.

Requirements:

Implement high-performance ML algorithms and distributed training infrastructure optimized for massive-scale multi-omic datasets.
Design and develop robust experimentation pipelines to enable rapid iteration, precise evaluations, and reproducible research outcomes.
Refactor research prototypes into clean, maintainable, and production-grade repositories suitable for large-scale deployments.
Direct ownership of research direction and model architecture decisions within a flat organization to influence the company's trajectory.

Jack & Jill

About Jack & Jill

Ok, I'll go first. I'm Jack, an AI that gets to know you deeply, what you’re great at and what you want out of your career. I then work to make that a reality – finding the best fit public job listings as they come up, supporting with applications, interview prep, and moral support. I also work together with my partner Jill to find off-market opportunities and introduce you directly to companies, just like a recruiter would. And I'm Jill, an AI that spends time with hiring managers and talent teams to really understand their company, their open roles, and who will thrive there. I work with Jack to find great fit people, like a headhunter with 1000x the network. Head to website and have a chat with us, we're always online.

Founded: 2018

Company size: 1 - 10

Website LinkedIn See all jobs →

Job description

This is a job that Jill, our AI Recruiter, is recruiting for on behalf of one of our customers.
She will pick the best candidates from Jack's network

The next step is to speak to Jack.

Machine Learning Research Engineer

Company Description: Prima Mente - Frontier Biology AI Lab

Job Description:
You will design and scale biological foundation models to decode the human brain. Training 7B+ parameter models on trillions of tokens, you will bridge high-level research and production-grade systems. This role is central to developing AI that detects neurological diseases years earlier, directly translating frontier machine learning into life-changing clinical outcomes.

Location: San Francisco, USA / London, UK

Why this role is remarkable:

Work at the absolute frontier of AI and biology, training 7B+ parameter models on 1.9T tokens of epigenomic data using a cluster of 256 H200 GPUs.
Direct ownership of research direction and model architecture decisions in a flat organization where technical excellence profoundly shapes the company’s trajectory.
Join a mission-critical lab solving the world's most complex biological system—the brain—to protect against Alzheimer's and other neurological diseases.

What you will do:

Implement high-performance ML algorithms and distributed training infrastructure optimized for massive-scale multi-omic datasets.
Design and develop robust experimentation pipelines to enable rapid iteration, precise evaluations, and reproducible research outcomes.
Refactor research prototypes into clean, maintainable, and performant production-grade repositories suitable for large-scale deployments.

The ideal candidate:

Extensive experience training and deploying large-scale models (7B+ parameters) with deep expertise in PyTorch, JAX, or TensorFlow.
Strong background in low-level optimization including quantization, CUDA/Triton, and hardware-specific tuning for GPU/TPU/HPU clusters.
Proven track record of handling massive datasets (2T+ tokens) and building scalable data processing workflows for high-throughput training.

Who are Jack & Jill?
Ok, I'll go first. I'm Jack, an AI that gets to know you on a quick call, learning what you're great at and what you want from your career. Then I help you land your dream job by finding unmissable opportunities as they come up, supporting you with applications, interview prep, and moral support.
And I'm Jill, an AI Recruiter who talks to companies to understand who they're looking to hire. Then I recruit from Jack's network, making an introduction when I spot an excellent candidate.

Next steps
Step 1. Visit our website.
Step 2. Click 'Talk to Jack'.
Step 3. Talk to Jack so he can understand your experience and ambitions.
Step 4. Jack will make sure Jill (the AI agent working for the company) considers you for this role.
Step 5. If Jill thinks you're a great fit and her client wants to meet you, they will make the introduction.
Step 6. If not, Jack will find you excellent alternatives. All for free.

We never post fake jobs
This isn't a trick. This is an open role that Jill is currently recruiting for from Jack's network.
Sometimes Jill's clients ask her to anonymize their jobs when she advertises them, which means she can't share all the details in the job description.
We appreciate this can make them look a bit suspect, but there isn't much we can do about it.
Give Jack a spin! You could land this role. If not, most people find him incredibly helpful with their job search, and we're giving his services away for free.