Principal ML Engineer - Large Scale Training Performance Optimization (JAX)

your career. Principal Large Scale Training Performance Optimization Engineer THE ROLE: We are looking for a Principal..., Expert Parallel ZeRO), and be familiar with training large models at scale. KEY RESPONSIBILITIES: Train large models...