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.
Similar Jobs
What We Do
Jack & Jill AI is an AI-powered recruiting platform that automates the job search process, offering assistance with applications, interview preparation, and job matching. It also helps companies find suitable candidates.
