Member of the Technical Staff, Pretraining

About the role

Output is revolutionizing biological reasoning through its advanced AI model, capable of understanding complex biological systems and developing novel drug treatments. As a key member of the team, you will be instrumental in advancing the core architecture and training of the foundation model, driving breakthroughs in biological reasoning.

Responsibilities

• Design and optimize architectures and training objectives for the foundation model
• Develop methods for learning across multiple biological data modalities
• Predict and understand biological phenomena using the model's reasoning capabilities
• Own the pretraining process, from experiment design to distributed training on multi-GPU clusters
• Evaluate the model's performance rigorously, ensuring it learns real biological reasoning

Requirements

• PhD in computer science, machine learning, physics, mathematics, or a related field with 2+ years of post-doctoral or industry research experience
• Strong publication record at top-tier venues in representation learning and model pretraining
• Experience pretraining large models on diverse, heterogeneous data
• Proficiency in Python and PyTorch, with experience in distributed multi-GPU infrastructure
• Experience owning the entire research-to-training pipeline

Qualifications

• Hands-on experience in representation learning and model pretraining
• Ability to design and execute rigorous evaluations
• Strong coding skills and experience in production-quality software development

Skills

• Representation learning
• Pretraining methods
• Self-supervised learning
• Foundation models
• Multimodal learning
• Heterogeneous data sources

Benefits

• Healthcare benefits
• Equity in a growing, well-funded startup

Pay

• Competitive salary

Schedule

• Full-time

Our Values

• Heart: Ownership and passion
• Excellence: Commitment to high standards
• Practicality: Results-oriented thinking
• Honesty: Direct communication and transparency
• Fun: A fun, engaging, and rewarding work environment

What We Offer

• Healthy feedback and support
• Competitive salary and equity
• Excellent medical, dental, and vision coverage