Principal ML Engineer
Vancouver, BCElectronic Arts
20+ YOECandidates need 20+ years of development experience, including production ownership of distributed-training data input. They need expertise in custom PyTorch or similar stacks, FSDP and parallelism, multimodal streaming tools such as Ray Data or NVIDIA DALI, and profiling of I/O, CPU, memory and accelerators.
PyTorch, TorchTitan, JAX, MaxText, Megatron Core, NeMo