Senior Machine Learning Research Engineer
Core
Build and scale next-generation generative and predictive models of cellular behaviour using machine learning to drive disease understanding and drug discovery.
Role type
Senior Machine Learning Research Engineer (IC)
Builds
Production-grade ML systems for large, multi-modal biological data applied to experimental biology
Domain
Biotech / Drug Discovery / Single-cell multi-omics
Deliverable
production ML models
Required skills
Python, PyTorch, distributed training frameworks, GPU workload profiling and optimization, cloud-based ML infrastructure, containerized environments, translating research prototypes to robust infrastructure
Preferred skills
CUDA / Triton kernel development, FlashAttention-style attention implementations, foundation models for biology/vision/language, open-source ML framework contributions
Technologies
PyTorch, CUDA, Triton, multi-GPU, multi-node configurations
Responsibilities
Implement and optimize large neural network models; profile and optimize training across compute, memory, and I/O; design and implement distributed training strategies; build and maintain core ML infrastructure; optimize inference and downstream deployment; address numerical, performance, and reliability issues across the stack; establish and maintain engineering practices in research code
Seniority
Senior, hands-on IC