🛠 Tech Stack

💼 About This Role

You'll research, develop, and scale large multimodal transformer models for drug discovery. Your work will directly impact real therapeutic decisions by putting advanced ML into scientists' hands. This role offers flexibility across career levels.

🎯 What You'll Do

• Research and implement architectural improvements for multimodal transformers.
• Optimize training pipelines for efficiency and scalability.
• Develop inference optimization techniques for interactive workflows.
• Design benchmarking and evaluation frameworks across modalities.

📋 Requirements

• MS in ML/CS or computational STEM field with relevant experience, or PhD.
• Strong Python and PyTorch experience for deep learning models.
• Demonstrated experience training transformer models at scale.
• Comfort with modern ML infrastructure (Docker, CUDA, Kubernetes, W&B).

✨ Nice to Have

• Experience with multimodal or multi-task model architectures.
• Training and inference optimization (mixed precision, kernel optimization, quantization).
• Familiarity with biomedical, chemical, or biological data domains.

🎁 Benefits & Perks

• 🏖️ Uncapped vacation
• 💰 Competitive pay
• 🏥 Company paid healthcare
• 🏋️ Onsite gym
• 🍽️ Onsite dining

📨 Hiring Process

Estimated timeline: 2-4 weeks · AI estimate

1. 1Recruiter Call· 30 min
2. 2Technical Interview· 60 min
3. 3On-site Interview· Half day