SIGNUP
Back to Jobs
XX
Machine Learning Research Engineer at Prima MenteJack and Jill AI • San Francisco, California, United States
Find similar jobs

This job offer is no longer available

XX

Machine Learning Research Engineer at Prima Mente

Jack and Jill AI
  • US
    San Francisco, California, United States
  • US
    San Francisco, California, United States
Find similar jobs

About

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.
  • San Francisco, California, United States

Languages

  • English
Notice for Users

This job was posted by one of our partners. You can view the original job source here.

Find similar jobs