Principal ML Engineer

Electronic Arts Ltd; Redwood City (CA); Presencial; USD 227 000 - 345 000; Tempo integral

Vantagens oferecidas por esta oferta de emprego

Paid time off

Sick time

Holidays

Family leave

Medical/dental/vision insurance

Life insurance

Disability insurance

401(k)

Resumo da oferta

Electronic Arts seeks a Principal ML Engineer, Training Pipelines, to own and shape the data paths that feed large-scale multimodal model training across diverse game data. You will build robust input pipelines, schemas, and deterministic loaders while leading a team and collaborating with researchers to translate evolving requirements into scalable infrastructure.

You will design and optimize batch composition, data quality, and reproducibility, while benchmarking throughput, latency, and

Qualificações

• 20+ years of development experience in distributed-training input systems with measurable improvements.
• Expertise building custom PyTorch training/ingestion systems or adapting TorchTitan, JAX/XLA with MaxText, Megatron Core/NeMo.
• Expertise in FSDP/HSDP, tensor/context/pipeline parallelism, mixed-precision training on GPU/TPU/AWS Trainium.
• Pre-training experience and understanding of model architecture, data representation, batching, and orchestration impact on efficiency.
• Expertise in large-scale multimodal streaming or hardware-accelerated ingestion with Ray Data, Mosaic Streaming, WebDataset, NVIDIA DALI, NVDEC.
• Deep knowledge of sharding, sampling, packing, caching, prefetching, deterministic data mixing, and reproducible training/evaluation.
• Experience profiling I/O, CPU, memory, network, distributed workers, and accelerators, including cross-layer failures

Responsabilidades

• Architect and build shared training input pipelines from game capture to accelerators, with storage, decoding, sharding, sampling, packing, batching, caching, prefetching, and host-to-device delivery across frameworks.
• Define data schemas, contracts, encodings, compression, and standards for dataset versioning, lineage, quality, and reproducible splits.
• Build deterministic loaders and checkpoint-safe iteration that recover from failures and support elastic restarts and changes in cluster size.
• Design reproducible dataset mixtures and balanced batches across modalities, resolutions, sequence lengths, and capture conditions.
• Benchmark throughput and accelerator utilization; model FLOPs utilization and scaling efficiency across thousands of accelerators.
• Diagnose bottlenecks across storage, network I/O, CPU, memory, decoding, preprocessing, and accelerators; handle stragglers and faulty hardware.
• Lead training-pipeline architecture reviews, mentor engineers, and align priorities with research leads; translate requirements into scalable shared infrastructure.

Conhecimentos

Distributed training

PyTorch

JAX/XLA

Megatron Core/NeMo

FSDP/HSDP

Tensor/pipeline parallelism

Mixed-precision training

Data ingestion systems

Ferramentas

Ray Data

Mosaic Streaming

WebDataset

NVIDIA DALI

NVDEC

Descrição da oferta de emprego

Electronic Arts creates next-level entertainment experiences that inspire players and fans around the world. Here, everyone is part of the story. Part of a community that connects across the globe. A place where creativity thrives, new perspectives are invited, and ideas matter. A team where everyone makes play happen.

At EA, we believe games are powerful because they bring together multiple ways people engage: play, watch, create, and connect. And increasingly, the biggest entertainment platforms aren't just places to consume content—they're places where communities build.

Creator-made content is already a proven part of EA's history—from community creation tools in Battlefield to The Gallery in The Sims 4. We believe new creative technologies and tools will expand how players engage with and contribute to our experiences, supported by thoughtful product design, safety systems, and global reach. Our focus is on enabling more players to participate in creative expression by making creation easier, safer, and more rewarding.

As a Principal ML Engineer, Training Pipelines, you will own the systems that prepare and deliver multimodal game data for large-scale model training. You will set the architecture for these training input pipelines and improve data throughput, accelerator utilization, reproducibility, and recovery from failures.

Your responsibilities will initially focus on hands-on individual contribution and technical leadership, with potential to take on people-management responsibilities as the team grows. You will partner with researchers and central data and compute teams to turn evolving model requirements into reusable training capabilities.

Responsibilities:

• You will architect and build shared training input pipelines from certified game capture to accelerators, covering storage layout, decoding, sharding, sampling, packing, batching, caching, prefetching, and host-to-device delivery across multiple training frameworks.
• You will define schemas, data contracts, encodings, compression, and standards for dataset versioning, lineage, quality, coverage, acceptance criteria, and reproducible splits.
• You will build deterministic loaders and checkpoint-safe iteration that recover from worker or hardware failures and corrupt data, including elastic restarts and changes to cluster size or parallelism.
• You will design reproducible dataset mixtures and balanced batches across modalities, resolutions, sequence lengths, titles, and capture conditions, accounting for worker/rank locality and distributed parallelism.
• You will benchmark and enforce throughput, data-wait time, step time, and accelerator utilization and stalls. Yu will model FLOPs utilization (MFU) where applicable, per-rank skew, scaling efficiency, restart time, reliability, and cost targets through thousands of accelerators.
• You will diagnose bottlenecks and intermittent failures across storage, network I/O, CPU, memory, decoding, preprocessing, and accelerators, including stragglers, faulty hardware, corrupt samples, and imbalanced batches.
• You will integrate compilers, runtimes, optimized kernels, low-precision training, and hardware-accelerated media processing. You will maintain efficient data delivery and recovery under FSDP/HSDP, tensor, context, and pipeline parallelism across heterogeneous accelerator environments.
• You will lead training-pipeline architecture reviews, mentor engineers, and align priorities with research leads. You will translate model and data-access requirements into scalable systems building on shared infrastructure services, while researchers retain model-research ownership.

Qualifications:

• 20+ years of development experience, including owning production systems for distributed-training input, with measurable improvements in throughput, accelerator utilization, step time, reliability, recovery time, or cost
• Expertise building custom PyTorch training and ingestion systems or adapting TorchTitan, JAX/XLA with MaxText, Megatron Core/NeMo, or equivalent stacks for single jobs spanning thousands of accelerators
• Expertise in FSDP2/HSDP, tensor/context/pipeline parallelism, mixed-precision training, and optimization on one or more accelerator platforms such as GPUs, TPUs, or AWS Trainium
• 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.
• Deep knowledge of sharding, sampling, packing, caching, prefetching, deterministic data mixing, checkpoint-safe iteration, and elastic restart; experience with schemas, dataset versioning, lineage, quality, and reproducible training/evaluation
• Experience profiling I/O, CPU, memory, network, distributed workers, and accelerators, including diagnosing cross-layer failures and intermittent faults

Pay Transparency - North America

COMPENSATION AND BENEFITS

The ranges listed below are what EA in good faith expects to pay applicants for this role in these locations at the time of this posting. If you reside in a different location, a recruiter will advise on the applicable range and benefits. Pay offered will be determined based on a number of relevant business and candidate factors (e.g. education, qualifications, certifications, experience, skills, geographic location, or business needs).

PAY RANGES

* British Columbia (depending on location e.g. Vancouver vs. Victoria) *$204,600 - $287,500 CAD

* California (depending on location e.g. Los Angeles vs. San Francisco) *$227,300 - $344,700 USD

Pay is just one part of the overall compensation at EA.

In the US, we offer a package of benefits including paid time off (3 weeks per year to start), 80 hours per year of sick time, 16 paid company holidays per year, 10 weeks paid time off to bond with baby, medical/dental/vision insurance, life insurance, disability insurance, and 401(k) to regular full-time employees. Certain roles may also be eligible for bonus and other incentive programs.

For Canada, we offer a package of benefits including vacation (3 weeks per year to start), 10 days per year of sick time, paid top-up to EI/QPIP benefits up to 100% of base salary when you welcome a new child (12 weeks for maternity, and 4 weeks for parental/adoption leave), extended health/dental/vision coverage, life insurance, disability insurance, retirement plan to regular full-time employees. Certain roles may also be eligible for bonus and other incentive programs.

About Electronic Arts

We’re proud to have an extensive portfolio of games and experiences, locations around the world, and opportunities across EA. We value adaptability, resilience, creativity, and curiosity. From leadership that brings out your potential, to creating space for learning and experimenting, we empower you to do great work and pursue opportunities for growth.

We adopt a holistic approach to our benefits programs, emphasizing physical, emotional, financial, career, and community wellness to support a balanced life. Our packages are tailored to meet local needs and may include healthcare coverage, mental well-being support, retirement savings, paid time off, family leaves, complimentary games, and more. We nurture environments where our teams can always bring their best to what they do.

Electronic Arts is an equal opportunity employer. All employment decisions are made without regard to race, color, national origin, ancestry, sex, gender, gender identity or expression, sexual orientation, age, genetic information, religion, disability, medical condition, pregnancy, marital status, family status, veteran status, or any other characteristic protected by law. We will also consider employment qualified applicants with criminal records in accordance with applicable law. EA also makes workplace accommodations for qualified individuals with disabilities as required by applicable law.