Machine Learning Research Engineer at Frontier AI biology lab

About this role

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:
Frontier AI biology lab


Job Description:
You will design and scale foundational AI models for multi-omics at massive scale. By implementing high-performance ML algorithms and maintaining robust experimentation pipelines, you will drive breakthroughs in scientific understanding. This role bridges the gap between AI research and clinical outcomes, focusing on understanding and protecting the human brain through technology.


Location:
San Francisco, USA; London, UK


Why this role is remarkable:

• Contribute to research infrastructure powering discoveries that could impact millions of lives globally.
• Work at the absolute frontier of AI and biology with high autonomy to propose state-of-the-art solutions.
• Join a multidisciplinary team of world-class AI researchers and clinicians in a well-funded research environment.

What you will do:

• Implement and optimize high-performance ML algorithms for reliability and massive scalability.
• Design and maintain experimentation pipelines for rapid iteration and reproducible research outcomes.
• Refactor prototype code into production-grade repositories and develop high-speed data processing workflows.

The ideal candidate:

• Deep expertise in modern ML frameworks like PyTorch or JAX and distributed computing principles.
• Proven experience training, optimizing, and deploying large-scale models with over 7B parameters.
• Skilled in low-level algorithm optimization including quantization, CUDA, or hardware-specific tuning.

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 f...