Member of Technical Staff, Generative

Output Labs; San Francisco (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Competitive salary and equity

Excellent medical, dental, and vision

Job summary

Output Labs is building a biological reasoning model capable of designing and evaluating novel molecules across multiple modalities. You will lead design, training, and evaluation of generative architectures, and run experiments on distributed GPU clusters.

You will own the end-to-end pipeline from research ideas to shipped models, writing production-quality code and contributing to an open, fast-moving biotech startup.

Qualifications

• PhD in CS/ML/Physics/Math or related field with post-doc or industry research experience.
• Strong publication record in generative methods at top-tier venues (e.g., NeurIPS, ICML, ICLR).
• Hands-on experience designing, building, and training deep generative models.
• Proficient in Python and PyTorch; experience training models on distributed multi-GPU infrastructure.
• Ability to own the full research-to-training pipeline: you train and ship models.

Responsibilities

• Design and build generative architectures for molecular data spanning multiple modalities.
• Develop training approaches that learn from diverse biological signals and enable novel structures.
• Build controllable generation methods meeting biological constraints while respecting chemistry rules.
• Integrate biological reasoning from the foundation model into the generative pipeline.
• Own training end-to-end: experiment design, distributed multi-GPU training, hyperparameter tuning.
• Design evaluation frameworks to measure biological meaningfulness, structural validity, and novelty.

Skills

Python

PyTorch

Distributed GPUs

Research-to-training

Experiment design

Education

PhD in CS/ML/Physics/Math (2+ yrs post-doc)

Bachelor's/Master's with 5+ yrs hands-on research

Tools

Git

Docker

Job description

Output has built a biological reasoning model that understands biology at the scale and complexity life actually operates. Our model independently learned the principles of molecular interactions, opening up drug treatments that were previously impossible. We're already generating therapies that traditional approaches cannot reach. The hardest problems in both AI and biology are being solved here, and there is room for you to own one.

Output is currently in stealth, operated by a team of repeat founders and biotech veterans with multiple exits in AI x Bio, and backed by top-tier VCs including Y Combinator.

You will lead the design and development of Output's generative models, working across molecular modalities to build systems that produce novel, biologically grounded molecules. This role spans the full arc from research to trained model: you design architectures, develop training approaches, run experiments on distributed GPU clusters, and evaluate results.

• You will design and build generative architectures for molecular data spanning multiple modalities, including small molecules, peptides, mini proteins and more

• You will develop training approaches that learn from diverse biological signal, ensuring the model composes genuinely novel structures

• You will build methods for controllable, targeted generation, enabling the model to produce molecules with specified biological properties while satisfying real-world chemical constraints

• You will integrate biological reasoning from your foundation model into the generative pipeline, using learned biological representations to guide and condition generation

• You will own training end-to-end: experiment design, distributed training on multi-GPU clusters, hyperparameter optimization, and iteration

• You will design evaluation frameworks that go beyond statistical metrics to measure whether generated molecules are biologically meaningful, structurally valid, and genuinely novel

About You

• You have a PhD in computer science, machine learning, physics, mathematics, or a related field with 2+ years of post-doctoral or industry research experience, or a Bachelor's or Master's degree with 5+ years of hands-on research and engineering experience in generative modeling

• You have a strong publication record in generative methods at top-tier venues (e.g., NeurIPS, ICML, ICLR)

• You have extensive hands-on experience designing, building, and training deep generative models, including work on novel architectures, training objectives, or sampling methods

• You are proficient in Python and PyTorch, and have experience training models on distributed multi-GPU infrastructure

• You have demonstrated the ability to own the full research-to-training pipeline: you do not just design methods, you train and ship models

• You write production-quality code that is well-tested and maintainable, and you are comfortable working in shared codebases with version control and code review

• You are a rigorous experimentalist who designs evaluations carefully, tracks experiments systematically, and draws conclusions from data

Bonus Points

• You have experience applying generative models to molecular, chemical, or biological data

• You have a background in chemistry, biology, computational biology, biophysics, or a related natural science

• You have experience with multi-modal learning or cross-modality translation

• You have experience with conditional or controllable generation methods

• You have contributed to open-source machine learning projects

Our Values

Heart: We foster a culture of ownership. We are assembling a team of individuals who are passionate and take pride in their contributions.

Excellence: We have an unwavering commitment to excellence and continuously challenge ourselves to reach the highest standards.

Practicality: We value practicality and results-oriented thinking. We are committed to making a tangible impact on the lives of patients and the broader community.

Honesty: We place a high value on honesty and directness. We firmly believe in addressing issues as they arise, in an open and transparent manner.

Fun: We believe that life is too short to not have fun. Our goal is to create a workplace that is fun, engaging, rewarding and fulfilling.

What We Offer

• We encourage new and different ideas, creativity and contrarian thinking

• Healthy feedback focused environment to help you strive - leadership will have high expectations, regularly share constructive feedback, support you and help you grow, and welcome receiving feedback and ideas from you

• You own your day-to-day management. What we care about is that we all hit our milestones

• Competitive salary and equity in a growing, well-funded startup

• Excellent medical, dental, and vision coverage