Machine Learning Engineer; NYC

The Role

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 Engineer, you'll work alongside our founders and team members to develop and implement cutting‑edge AI systems capable of complex biological reasoning across multiple scales.

• You will build foundational models for biology capable of reading and writing biology at scale
• You will develop deep generative models for biological applications, exploring innovative architectures to capture the complexities of multi‑scale biological systems
• You will work on distributed training systems to scale our models to billions of parameters, optimizing for performance and efficiency across multi‑GPU and multi‑node setups while handling large‑scale biological datasets
• You will engineer efficient data pipelines to manage and process massive biological datasets, addressing challenges in data loading, splitting, and memory optimization
• You will develop and implement robust evaluation frameworks for complex biological models, ensuring data integrity and preventing leakage across dataset splits

• You have a Bachelor's in Computer Science, Machine Learning, or a related technical field
• You have 3+ years of experience in developing and implementing deep generative learning models
• You have experience pre‑training models and are proficient in distributed computing environments
• You are proficient in Python and have expertise in at least one major deep learning framework (PyTorch, Tensor Flow, or JAX)
• You have experience with deep learning and generative architectures such as transformers, diffusion models and autoencoders
• You are skilled in working with terra‑scale datasets and scaling models to billions of parameters
• You have a strong understanding of machine learning fundamentals, including various model architectures, optimization techniques, and evaluation metrics
• You have experience in designing and implementing efficient data pipelines for processing and managing large datasets
• You are experienced in developing robust evaluation frameworks and ensuring data integrity in machine learning projects
• You are experienced in code organization, version control, and collaborative software development practices.

• 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

• You have experience applying machine learning to biology or chemistry
• You have contributed to open‑source machine learning projects or published research papers in the field of AI/ML
• You have experience optimizing machine learning models for high‑performance computing environments
• You are familiar with ML‑Ops practices and tools for managing ML experiments and deployments

• ❤️
  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.

• 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