GPU ML Engineer for Apple Silicon Acceleration

MaziCTools; Los Angeles (CA); Ibrido; USD 150.000 - 210.000; Tempo pieno

Descrizione del lavoro

Apple in Cambridge, Massachusetts seeks extraordinary machine learning and GPU programming engineers to provide robust compute solutions for accelerating ML networks on Apple Silicon.

The role offers opportunities to influence design of compute and programming models in next-generation GPU architectures and to contribute to ML training acceleration technologies across multiple platforms.

Competenze

• Strong programming and problem-solving skills.
• Solid understanding of machine learning fundamentals.
• Knowledge of GPU compute programming models and optimization techniques.
• Experience with GPU compute framework development and optimization.
• Experience with ML frameworks (TensorFlow, PyTorch or JAX).

Mansioni

• Develop ML training acceleration techniques for GPU.
• Implement/optimize APIs in ML frameworks and Graph APIs.
• Tune GPU-accelerated training across products.
• Analyze performance and optimize compiler/kernel code.
• Support adoption of GPU-accelerated training across clients.

Conoscenze

Programming

Machine learning basics

GPU compute

API development

ML frameworks

Strumenti

LLVM/MLIR

Metal API

Descrizione del lavoro

Apple in Cambridge, Massachusetts seeks extraordinary machine learning and GPU programming engineers to provide robust compute solutions for accelerating ML networks on Apple Silicon.

The role offers opportunities to influence design of compute and programming models in next-generation GPU architectures and to contribute to ML training acceleration technologies across multiple platforms.