Technical Staff Member

Job Summary

A company is looking for a Member of Technical Staff, Post training team.

Key Responsibilities

• Design and write high-performant and scalable software for training models
• Consistently post-train models to achieve state-of-the-art performance
• Coordinate with specialist teams to produce models with strong overall performance

Required Qualifications

• Extremely strong software engineering skills
• Proficiency in Python and related ML frameworks such as JAX, Pytorch, and XLA/MLIR
• Experience with distributed training infrastructures (Kubernetes, Slurm) and frameworks (Ray)
• Hands-on experience with training large models at scale and post-training performance optimization
• Experience using large-scale distributed training strategies