Senior ML Ecosystem Engineer: GPU-Optimized ML Apps

Lex; Greater London; On-site; GBP 100,000 - 140,000; Full time

Job summary

Apple's Metal Pro Apps and ML Ecosystem team seeks an engineer to optimize ML workloads for ANE and GPU silicon in macOS and iOS apps. You will analyze how developers use the ML API, propose optimizations, contribute to open source projects, and host developer sessions to advance ML tooling and performance.

You will be a key advisor for ML app development, including engineering, documentation, training, and maintaining relationships with developers and open source communities.

Qualifications

• GPU compute framework development, maintenance, and optimization.
• Proven programming and problem-solving skills.
• Good understanding of machine learning fundamentals.
• Exemplary written and oral communication skills.

Responsibilities

• Analyze developers products and report on their use of the ML API, recommending optimizations.
• Contribute platform enablement and optimisations to key open source projects.
• Host developers during in-person engineering sessions.

Skills

GPU compute

Programming skills

ML fundamentals

Communication skills

Tools

PyTorch

TensorFlow

JAX

Job description

Apple's Metal Pro Apps and ML Ecosystem team seeks an engineer to optimize ML workloads for ANE and GPU silicon in macOS and iOS apps. You will analyze how developers use the ML API, propose optimizations, contribute to open source projects, and host developer sessions to advance ML tooling and performance.

You will be a key advisor for ML app development, including engineering, documentation, training, and maintaining relationships with developers and open source communities.