Principal Engineer Tech Lead Manager, ML Acceleration

We are seeking an experienced Principal Engineer Tech Lead Manager to build and lead our new Machine Learning (ML) Acceleration team. The role will drive the strategy, development, and execution of initiatives aimed at significantly accelerating ML model training.

Requirements

• 8+ years of experience in software engineering, with at least 5+ years focused on Machine Learning systems, infrastructure, or performance optimization.
• 3+ years of experience in a technical lead or management role, with a proven track record of building and leading high-performing teams.
• Extensive experience with large-scale ML model training and deployment, ideally in a production environment.
• Strong understanding of distributed systems and cloud computing platforms (e.g., AWS, GCP, Azure).
• Deep expertise in ML frameworks such as PyTorch or JAX.
• Proficiency in performance profiling and optimization techniques for ML workloads.
• Strong programming skills in Python; C++ experience is a plus.
• Knowledge of MLOps principles and practices.

Benefits

• Medical
• Dental
• Vision
• 401k with a company match
• Health saving accounts
• Life insurance
• Pet insurance