Machine Learning Engineer - Output Services : Job Details

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

Who We're Looking For:

• 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, TensorFlow, 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.

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 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

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!