Member of Technical Staff, Training Infra

You will design and optimize distributed training systems across large GPU fleets. You will improve training throughput and efficiency and develop reusable frameworks that strengthen reliability, reproducibility, and scalability for new model architectures.

Responsibilities

• Design and optimize distributed training systems across GPUs and nodes
• Develop high-performance optimizations for training throughput and efficiency
• Develop reusable training frameworks and libraries
• Improve training reproducibility reliability and scalability

Requirements

• PyTorch
• TensorFlow
• Engineering
• Python
• C++
• Rust
• Go
• Docker
• Kubernetes
• CI/CD

Benefits

• Equity
• Flexible vacation and paid time off
• Health insurance
• Dental insurance
• Vision insurance
• 401k match
• Catered meals
• Commuter subsidies