Altos Labs is hiring a Machine Learning Scientist

Altos Labs is looking for a Machine Learning Scientist to join our state-of-the-art Virtual Cell team. You will help accelerate the development of multi-modal generative foundation models for multiscale biology, working within multidisciplinary teams to improve 'virtual cell' models that advance our mission of discovering novel interventions for aging and disease.

What You'll Do

• Design, develop and evaluate state-of-the-art foundation and focused models, at scale, to advance the Altos mission.
• Pre-train and fine-tune large-scale machine learning systems using multimodal biological data and prior knowledge inputs.
• Pioneer novel machine learning methodologies and statistical frameworks (e.g., generative models, diffusion/flow matching models, and advanced transformer architectures) to address fundamental challenges in cell health and rejuvenation.
• Design, implement, and optimize large-scale machine learning systems using modern frameworks (e.g., PyTorch, JAX), AI-assisted coding, and agile practices.
• Develop and manage efficient distributed training strategies across multiple GPUs and compute clusters to handle terabytes of multi-modal biological data.
• Develop robust approaches for multi-modal data integration and cross-domain mapping to extract actionable biological insights.
• Participate in the full ML development lifecycle from theoretical conception and data strategy through model development, training, and evaluation.

What We're Looking For

• Ph.D. in Machine Learning, Computer Science, Artificial Intelligence, Statistics, or a related quantitative field, demonstrating a deep theoretical foundation in ML/AI.
• Relevant work experience in either an academic or industry setting.
• Prior experience in developing and implementing novel generative AI models in a subset of the following: transformers, multi-modality, diffusion/flow matching models.
• Deep understanding and expertise of Machine Learning Principles and how they apply to different models.
• Proven experience developing and applying complex machine learning models, preferably with a significant portion of that time spent in a fast-paced industry or translational research environment.
• Very strong programming skills, including experience with Python and deep learning libraries (PyTorch, Hugging Face Transformers, H-F Datasets, H-F Accelerate).
• 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.

Nice to Have

• Strong track record of published peer reviewed innovative AI/ML research.
• Experience in cell health and rejuvenation related research area.
• Experience in the application of machine learning methods to biological data.
• Experience in computational approaches to drug discovery.
• Experience with software development methodologies and open-source software.

Technical Stack

• Python
• PyTorch
• JAX
• TensorFlow
• Hugging Face Transformers
• H-F Datasets
• H-F Accelerate

Team & Environment

You will be part of the multidisciplinary Institute of Computation and the Virtual Cell team.

Benefits & Compensation

• Redwood City, CA: Scientist I $211,200 - $257,500, Scientist II $237,800 - $290,000, Senior Scientist I $270,600 - $330,000.
• San Diego, CA: Scientist I $188,600 - $230,000, Scientist II $223,900 - $273,000, Senior Scientist I $251,700 - $307,000.

Altos Labs provides equal employment opportunities to all employees and applicants for employment, without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws.