Machine Learning Engineer

The Role

Join Output Biosciences in revolutionizing our understanding of biology through cutting-edge generative AI. As a Machine Learning Engineer, you will collaborate with founders and team members to develop advanced AI systems, enabling complex biological reasoning from molecules to organisms.

Key Responsibilities

• Design and implement advanced machine learning algorithms to enhance model performance and biological understanding.
• Develop and fine-tune generative models for biological applications, exploring innovative architectures to capture multi-scale biological systems.
• Work on distributed training systems to scale models to billions of parameters, optimizing performance and efficiency across multi-GPU and multi-node setups.
• Engineer efficient data pipelines to manage and process massive biological datasets, addressing challenges in data loading, splitting, and memory optimization.
• Develop and implement robust evaluation frameworks for complex biological models, ensuring data integrity and preventing leakage across dataset splits.

Who We're Looking For

• Bachelor's degree in Computer Science, Machine Learning, or a related technical field.
• 3+ years of experience in developing and implementing machine learning models.
• Proficiency in Python and expertise in deep learning frameworks such as PyTorch, TensorFlow, or JAX.
• Experience with deep learning and generative architectures, including transformers, diffusion models, and autoencoders.
• Skilled in working with large-scale datasets and distributed computing environments.
• Experience with AWS for training, inference, and deployment.
• Strong understanding of machine learning fundamentals, model architectures, optimization techniques, and evaluation metrics.
• Experience in designing and implementing efficient data pipelines for processing large datasets.
• Experience in developing robust evaluation frameworks to ensure data integrity.
• Skilled in code organization, version control, and collaborative software development practices.
• Excellent problem-solving skills and the ability to quickly adapt to new challenges.
• Proactive approach to problem-solving, taking ownership of challenges with a mature mindset in ambiguous situations.
• Excellent communication skills for articulating complex technical concepts clearly.
• Motivated by making a real impact with determination and creativity.

Bonus Points

• Experience applying machine learning to biology or chemistry.
• Contributions to open-source machine learning projects or published research papers in AI/ML.
• Experience optimizing machine learning models for high-performance computing environments.
• Familiarity with ML-Ops practices and tools for managing ML experiments and deployments.

Our Values

❤️ Heart: We foster a culture of ownership, encouraging passion and pride in our contributions.

🏆 Excellence: We commit to the highest standards of excellence and continuously challenge ourselves to improve.

🚀 Practicality: We value results-oriented thinking, aiming to make a tangible impact on our patients and the community.

📣 Honesty: We prioritize honesty and directness, addressing issues openly and transparently.

🎮 Fun: We believe in an engaging and enjoyable work environment, making our workplace fun, rewarding, and fulfilling.

What We Offer

• Encouragement of creativity and contrarian thinking, shaping our culture with early team members.
• Flexible work management; focus on milestones rather than location and hours.
• Competitive salary and equity in a growing, well-funded startup.
• Excellent medical, dental, and vision coverage.
• A feedback-focused environment to help you grow, with high expectations, regular constructive feedback, and support from leadership.

Company Name: Output Biosciences

Job Title: Machine Learning Engineer