🛠 Tech Stack

💼 About This Role

You'll join Unity Vector to build the offline ML platform powering insight and AI-driven decisions across the company. Your work will directly impact large-scale model training and distributed training workflows. This role bridges research and production for a recent PhD graduate.

🎯 What You'll Do

• Build and maintain data pipelines for training datasets
• Contribute to infrastructure for distributed training (PyTorch, Ray)
• Work with workflow orchestration tools (Airflow, Flyte)
• Improve reproducibility through dataset validation and monitoring

📋 Requirements

• PhD in Computer Science, Machine Learning, or related field
• Foundation in ML systems or distributed systems
• Experience with Python and data-intensive workloads
• Familiarity with ML frameworks (PyTorch, TensorFlow) or distributed systems

✨ Nice to Have

• Experience with workflow orchestration (Airflow, Flyte)
• Exposure to large-scale data platforms (data lakes, warehouses)
• Publications in ML systems or distributed systems

🎁 Benefits & Perks

• 🏥 Comprehensive health, life, disability insurance
• 🚌 Commute subsidy
• 📈 Employee stock ownership
• 🏖️ Generous vacation and personal days
• 👶 Parental leave and family-care programs

📨 Hiring Process

Estimated timeline: 3-5 weeks · AI estimate

1. 1Recruiter Screen· 30 min
2. 2Technical Phone Screen· 45 min
3. 3On-site / Virtual Interview· 4 hours