Video-Generation ML Engineer (Distributed Training)

Apple, Inc.; Cupertino (CA); Presencial; USD 142 000 - 263 000; Tempo integral

Vantagens oferecidas por esta oferta de emprego

Stock programs

Discretionary bonuses

Relocation assistance

Education reimbursement

Medical benefits

Resumo da oferta

Apple, Inc. in California is seeking a machine learning engineer to develop video generation models using large generative architectures. You will design training recipes and run distributed multi-GPU training, then optimize models for efficient production deployment.

The role requires a Bachelor's degree with 3+ years in the field, hands-on experience with PyTorch or JAX, and strong Python software skills. Apple offers competitive compensation and benefits.

Qualificações

• Bachelor's degree in a technical field and at least 3 years of relevant industry experience.
• Experience with large-scale generative model training for video generation.
• Experience running distributed training across multi-node GPU clusters.
• Strong Python software engineering skills with PyTorch or JAX.

Responsabilidades

• Pre-train video generation models with architecture selection and training recipe design.
• Run, monitor, and debug large-scale distributed training and diagnose instabilities.
• Improve training efficiency with parallelism, mixed precision, checkpointing and hardware use.
• Enhance inference efficiency via distillation, sampling strategies, quantization.
• Adapt pre-trained models with fine-tuning and knowledge distillation.
• Build and maintain training/evaluation code with reproducibility in mind.
• Collaborate with product and research to translate requirements into tasks.

Conhecimentos

Python programming

PyTorch

JAX

Distributed training

Generative models

Formação académica

Bachelor's degree in Electrical Engineering, Computer Science, or related field

MS or PhD in EE/CS/CE (preferred)

Ferramentas

PyTorch

JAX

Descrição da oferta de emprego

Apple, Inc. in California is seeking a machine learning engineer to develop video generation models using large generative architectures. You will design training recipes and run distributed multi-GPU training, then optimize models for efficient production deployment.

The role requires a Bachelor's degree with 3+ years in the field, hands-on experience with PyTorch or JAX, and strong Python software skills. Apple offers competitive compensation and benefits.