Technical Staff Member

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

Jun 13, 2025 - 20:10

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