Machine Learning Research Engineer at Frontier AI biology lab

Machine Learning Research Engineer

Frontier AI biology lab

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.

San Francisco, USA;
London, UK

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

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

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