# Principal ML Engineer

**Company**: Electronic Arts
**Location**: Vancouver, British Columbia
**Work arrangement**: hybrid
**Experience**: senior
**Job type**: full-time
**Category**: Engineering
**Industry**: Technology
**Ticker**: EA
**Wikidata**: https://www.wikidata.org/wiki/Q173941

**Apply**: https://jobs.ea.com/ro_RO/careers/JobDetail/Principal-ML-Engineer/216087?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_08680110-4ea

## Description

Electronic Arts is seeking a Principal ML Engineer to own the systems that prepare and deliver multimodal game data for large-scale model training. The successful candidate will set the architecture for these training input pipelines and improve data throughput, accelerator utilization, reproducibility, and recovery from failures.

Responsibilities:

- Architect and build shared training input pipelines from certified game capture to accelerators

- Define schemas, data contracts, encodings, compression, and standards for dataset versioning, lineage, quality, coverage, acceptance criteria, and reproducible splits

- Build deterministic loaders and checkpoint-safe iteration that recover from worker or hardware failures and corrupt data

- Design reproducible dataset mixtures and balanced batches across modalities, resolutions, sequence lengths, titles, and capture conditions

- Benchmark and enforce throughput, data-wait time, step time, and accelerator utilization and stalls

- Diagnose bottlenecks and intermittent failures across storage, network I/O, CPU, memory, decoding, preprocessing, and accelerators

- Integrate compilers, runtimes, optimized kernels, low-precision training, and hardware-accelerated media processing

- Lead training-pipeline architecture reviews, mentor engineers, and align priorities with research leads

Qualifications:

- 20+ years of development experience, including owning production systems for distributed-training input

- Expertise building custom PyTorch training and ingestion systems or adapting TorchTitan, JAX/XLA with MaxText, Megatron Core/NeMo, or equivalent stacks

- Expertise in FSDP2/HSDP, tensor/context/pipeline parallelism, mixed-precision training, and optimization on one or more accelerator platforms

- Pre-training experience and an understanding of how model architecture, data representation, batching, and orchestration affect training-system design and efficiency

- Expertise building large-scale multimodal streaming or hardware-accelerated ingestion with Ray Data, Mosaic Streaming, WebDataset, NVIDIA DALI, NVDEC, or equivalent technologies

## Skills

### Required
- PyTorch
- TorchTitan
- JAX/XLA
- MaxText
- Megatron Core/NeMo
- FSDP2/HSDP
- tensor/context/pipeline parallelism
- mixed-precision training
- Ray Data
- Mosaic Streaming
- WebDataset
- NVIDIA DALI
- NVDEC

---

Source: [Apply at jobs.ea.com](https://jobs.ea.com/ro_RO/careers/JobDetail/Principal-ML-Engineer/216087?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)