Job Description
Altos Labs is seeking a Staff Machine Learning Engineer to contribute to the development of generative AI/ML models for multi-modal, multiscale biology. This role involves partnering with scientists to generate biological insights and develop novel therapies. The ideal candidate will thrive in a fast-paced, collaborative environment focused on scientific excellence.
Responsibilities:
- Partner with world-class scientists across Altos to help generate biological insights with the goal of developing novel therapies.
- Design and implement large-scale machine learning algorithms and systems applied to biological datasets.
- Train, evaluate, and optimize machine learning models at scale.
- Communicate effectively with internal and external collaborators to meet ambitious research and development goals.
Requirements:
- Proven track record leveraging machine learning to solve real-world problems.
- Expertise in one or more of the following: generative models, language models, computer vision, bayesian inference, causal reasoning & inference, transfer & multi-task learning, diffusion models, graph neural networks, active learning.
- Experience writing production-quality code with modern machine learning frameworks such as PyTorch, TensorFlow, JAX, or similar.
- Experience with multi-GPU and distributed training at scale.
- Masters or Ph.D. degree in a quantitative/computational field such as computer science, artificial intelligence, mathematics, statistics, physics, or computational biology, or equivalent experience.
- Very strong programming skills, including experience with Python and deep learning libraries such as PyTorch or JAX.
- Experience in large-scale distributed optimization of machine learning models across multiple GPUs and nodes.
Altos Labs offers:
- A prominent role in developing generative AI/ML models for multi-modal, multiscale biology.
- Opportunity to partner with world-class scientists.
- A collaborative and scientifically excellent culture.