Machine Learning Research Scientist, Multimodal

Job Description

Join our team and help build the world's first biological reasoning model. Work with us to build generative foundational models that decode biological systems across scales - from molecules to organisms - enabling us to predict, understand, and program living systems in ways never before possible.

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.

As a Machine Learning Research Scientist, you'll work alongside our founders and team members bridging biology and AI to develop AI systems capable of complex biological reasoning across multiple scales and data modalities.

You will train and scale innovative multimodal AI models on large, diverse biological datasets across distributed GPU clusters

You will innovate on cutting-edge machine learning algorithms to enhance model performance and biological understanding

You will design and conduct experiments to validate the model's capabilities across a wide range of biological phenomena, from molecular interactions to complex cellular processes and beyond

You will train and scale innovative AI models using large, diverse biological multi-modal datasets across distributed GPU clusters

You will solve challenging AI research problems at scale, tackling key aspects such as model architecture and scaling, pre-training and fine-tuning strategies, and rigorous evaluation methods

Your work will span various advanced generative AI architectures, including transformers, diffusion models and autoencoders, exploring and implementing these techniques to push the boundaries of biological AI

Who We're Looking For:

You have a Bachelor's degree in Computer Science, Computer Engineering, Mathematics, Physics or a relevant technical field

You have a PhD with 2 years of research experience or equivalent experience with 4 years of research or industry experience in Deep Learning, Artificial Intelligence or other relevant fields

You have a strong publication record in top-tier AI conferences or journals (e.g., NeurIPS, ICML, ICLR)

You are proficient in Python and have expertise in at least one major deep learning framework (PyTorch, TensorFlow, or JAX)

You have researched and developed deep learning and generative AI methods

Who You Are:

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

Bonus points

You have experience applying machine learning to biological or chemical problems

You have knowledge of physics, molecular biology, chemistry, computational biology, bioinformatics, or cheminformatics or related fields

You have experience with multi-modal learning

You have contributed to open-source machine learning projects

You have experience in developing efficient code in C/C++

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 our 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. That's why we're committed to creating an engaging and enjoyable work environment. Our goal is to create a workplace that is fun, rewarding and fulfilling.

What We Offer

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

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

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

We are looking forward to further shaping our culture with early team members. You'll have a chance to make your mark!

Compensation Range: $120K - $250K

Full-time