Logo
Jack & Jill/External ATS

Machine Learning Research Engineer at Frontier AI biology lab

Jack & Jill/External ATS, San Francisco, California, United States, 94199

Save Job

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.

#J-18808-Ljbffr