🛠 Tech Stack

💼 About This Role

You'll build the inference and training frameworks for an AI document processing platform, directly impacting how enterprises extract data from complex PDFs. You'll scale distributed ML training across multi-node GPU clusters and create observability across the stack.

🎯 What You'll Do

• Build and maintain training and inference stack for fast iteration
• Design systems for scaling training across multi-node, multi-GPU environments
• Scale distributed training and inference across large GPU clusters
• Develop benchmarks to identify bottlenecks in training and inference stacks

📋 Requirements

• Strong Python skills and systems engineering background
• Experience with Kubernetes and distributed training frameworks
• Ability to solve complex problems from first principles

✨ Nice to Have

• Experience at an early-stage or high-growth startup
• Contributions to open source training/inference stacks
• Excitement for distributed inference across hundreds to thousands of GPUs

🎁 Benefits & Perks

• 🏖️ Unlimited PTO
• 🍽️ Free daily lunch at office
• 🚗 Reimbursed transportation costs
• 🏥 Generous health insurance covering medical, dental, vision
• 💪 $150/month health and wellness budget

📨 Hiring Process

Estimated timeline: 2-4 weeks · AI estimate

1. 1Recruiter screen· 30 min
2. 2Technical interview· 60 min
3. 3Onsite interviews· 3 hours