Machine Learning Research Engineer at Prima Mente

San Francisco, CA - USA

Monthly Salary: Not Disclosed

Posted on: 8 days ago

Vacancies: 1 Vacancy

Department:

Marketing

Job Summary

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 Jacks 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 companys trajectory.
Join a mission-critical lab solving the worlds most complex biological systemthe brainto protect against Alzheimers 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 Ill go first. Im Jack an AI that gets to know you on a quick call learning what youre 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 Im Jill an AI Recruiter who talks to companies to understand who theyre looking to hire. Then I recruit from Jacks 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 youre 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 isnt a trick. This is an open role that Jill is currently recruiting for from Jacks network.
Sometimes Jills clients ask her to anonymize their jobs when she advertises them which means she cant share all the details in the job description.
We appreciate this can make them look a bit suspect but there isnt 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 were giving his services away for free.

Required Experience:

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 Jacks networkThe next step is to speak to Jack.Machine Learning Research EngineerCompany Description: Prima Mente - Frontier Biology AI LabJob Description:You will desi...