🛠 Tech Stack

💼 About This Role

You'll build and improve the models and ML systems that drive protein design efforts at an AI-first startup. You'll deploy large-scale generative models and develop infrastructure to enable faster, more confident research. This role offers significant ownership over the ML stack in a small, fast-moving team.

🎯 What You'll Do

• Build robust pipelines for automated model fine-tuning, alignment, and evaluation
• Design modular multi-model pipelines for protein design
• Develop scalable ETL pipelines for petabyte-scale protein data
• Optimize model training and inference code for throughput and resource utilization

📋 Requirements

• BS or MS in Computer Science, Machine Learning, or related field
• 3+ years of hands-on experience building and training ML models in PyTorch
• Strong Python and software engineering fundamentals
• Experience profiling, benchmarking, and optimizing ML model training and inference

✨ Nice to Have

• Familiarity with protein language models or computational biology
• Experience with GPU-level optimization (CUDA, Triton)
• Experience with distributed training (DDP, FSDP, multi-node GPU clusters)

🎁 Benefits & Perks

• 💰 Competitive compensation with equity participation
• 🏦 401(k) with strong employer match
• 🩺 Comprehensive health/dental/vision insurance
• 🏖️ Generous PTO policy
• 📈 Professional development opportunities

📨 Hiring Process

Estimated timeline: 2-4 weeks · AI estimate

1. 1Recruiter Call· 30 min
2. 2Technical Screen· 60 min
3. 3On-site Interviews· 3 hours