Machine Learning Systems Engineer

We are looking for a Machine Learning Systems Engineer to join our ML Acceleration team, responsible for core systems that enable researchers to train frontier models at scale.

Requirements

• Performance Profiling & Optimization: Utilize profiling tools to identify bottlenecks in data loading, gradient computation, and communication.
• Distributed Training: Optimize distributed training pipelines using frameworks such as PyTorch Distributed.
• Kernel Development: Design and maintain high-performance GPU kernels in Triton or CUDA for state-of-the-art ML workloads.
• Data Pipeline Engineering: Optimize robust data loading pipelines that maximize training throughput.

Benefits

• Medical
• Dental
• Vision
• 401k with company match
• Health Saving Accounts
• Life Insurance
• Pet Insurance