🛠 Tech Stack

💼 About This Role

You'll research and build solutions across algorithms, scaling laws, data processing, optimizers, and model architecture for open-weight AI models. You'll design and run scientific experiments to advance large language model scaling and contribute to the full stack from low-level optimizations to high-level model design.

🎯 What You'll Do

• Design and run scientific experiments on scaling laws and data efficiency
• Implement state-of-the-art deep learning methods from literature
• Optimize training infrastructure for efficient scaling
• Lead small research projects independently and collaborate on larger initiatives

📋 Requirements

• Graduate degree (MS or PhD) in Computer Science, Machine Learning, or related discipline
• Solid software engineering with experience building large-scale systems
• Experience with large-scale ETL workflows and training data preparation
• Deep understanding of large-scale ML especially language models and distributed training

✨ Nice to Have

• Familiarity with PyTorch deep learning framework
• Experience with optimizers and model architecture innovations

🎁 Benefits & Perks

• 💰 Top-tier compensation and equity
• 🏥 Comprehensive medical, dental, vision, life, and disability insurance
• 👶 Fully paid parental leave for all new parents
• 🍽️ Daily lunch and dinner provided
• ✈️ Paid time off and relocation support

🚩 Heads Up

• Job title 'Member of Technical Staff' typically implies senior level but no years of experience stated
• Listing requires both research and engineering skills which may lead to role ambiguity