Member of Technical Staff (GPU Engineer)

Overview

Role involves designing and optimizing large-scale training infrastructure for AI models.

Ideal candidate has strong engineering skills in Python and experience with GPU programming.

Locations

• Singapore
• United States
• United Kingdom

Requirements

• Experience with Python
• Experience with GPU code (CUDA, Triton)
• Experience with large-scale compute clusters (Slurm, Kubernetes)

Responsibilities

• Design and implement improvements to training infrastructure
• Optimize models' training performance
• Collaborate with engineers and researchers

Benefits

• 5 weeks paid leave
• comprehensive healthcare benefits
• visa assistance