Build the world's first biological reasoning model and develop generative foundational models that decode biological systems across scales - from molecules to organisms - enabling prediction, understanding, and programming of living systems.
Requirements
- You have 3+ years of experience in developing and implementing deep generative learning models
- You have experience pre-training models and are proficient in distributed computing environments
- You are proficient in Python and have expertise in at least one major deep learning framework (PyTorch, TensorFlow, or JAX)
- You have experience with deep learning and generative architectures such as transformers, diffusion models and autoencoders
- You are skilled in working with terra-scale datasets and scaling models to billions of parameters
- You have a strong understanding of machine learning fundamentals, including various model architectures, optimization techniques, and evaluation metrics
- You have experience in designing and implementing efficient data pipelines for processing and managing large datasets
Responsibilities
- You will build foundational models for biology capable of reading and writing biology at scale
- You will develop deep generative models for biological applications, exploring innovative architectures to capture the complexities of multi-scale biological systems
- You will work on distributed training systems to scale our models to billions of parameters, optimizing for performance and efficiency across multi-GPU and multi-node setups while handling large-scale biological datasets
- You will engineer efficient data pipelines to manage and process massive biological datasets, addressing challenges in data loading, splitting, and memory optimization
- You will develop and implement robust evaluation frameworks for complex biological models, ensuring data integrity and preventing leakage across dataset splits
Other
- You have excellent problem-solving skills and the ability to quickly adapt to new challenges
- You exhibit a proactive approach to problem-solving, thinking beyond the specific task, taking ownership of challenges, and pride in solving them
- You have a mature mindset in ambiguous situations, helping to frame questions and seek clarity while making decisions in the face of uncertainty
- You have excellent communication skills and can clearly articulate complex technical concepts
- You are motivated by making a real impact and are committed to tackling problems of significant consequence with determination and creativity
