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.
Back to all jobs
See Open Roles
Learn about Jack
This job is no longer actively hiring. Open Roles to see active jobs.
Machine Learning Research Engineer at Primamente
Join Prima Mente, a frontier biology AI lab, to build foundation models that decode the human brain. We are training 7B+ parameter models on 1.9T tokens using 256 H200 GPUs to detect neurological diseases years before symptoms appear. As an ML Research Engineer, you will own model architecture and low-level optimizations, bridging the gap between cutting-edge research and clinical impact. If you want to apply large-scale ML systems to the most complex challenge in biology while working alongside world-class researchers in SF or London, this is your opportunity to change the future of human health.
Want to apply for this role?
This role is no longer actively hiring, but Jack can still help you discover similar open roles that fit.
Location
San Francisco, United States
Compensation
Not Disclosed
Company
Primamente
Role overview
Healthcare
Prima Mente is a frontier biology AI lab.
We generate our own data, build general purpose biological foundation models, and translate discoveries into research and clinical outcomes. Our first goal is to tackle the brain: to deeply understand it, protect it from neurological disease, and enhance it in health. Our team of AI researchers, experimentalists, clinicians, and operators is based in London, San Francisco and Dubai.
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.
Who this is a fit for
- 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.
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.
How Jack & Jill work together
I get to know what you’re great at, then find roles you’d never find yourself.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.
I recruit from Jack’s network and make the intro when I spot a great match.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.
Jack gets to know what you're great at and what you want next, then searches 15 million jobs daily and helps you discover roles at companies like this.
Meet Jack
What happens next?
Jack’s an AI agent for job searching and career coaching. He works for you.
Jill is the AI recruiter working for the company. She recruits from Jack’s network.
If your profile’s a match and Primamente wants to meet, Jill will make the intro. In the meantime, Jack will send you excellent alternatives.