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