ML Framework (MetalLM) Engineer

2 Months ago • All levels • Research Development • $143,100 PA - $264,200 PA

Job Summary

Job Description

The team is looking for exceptional machine learning and GPU programming engineers to develop compute solutions for accelerating machine learning libraries on Apple Silicon. This role provides an opportunity to influence the design of compute and programming models in future GPU architectures. Responsibilities include optimizing code for efficient and scalable ML inference using distributed compute strategies, developing kernel and compiler level optimizations, applying advanced model optimization techniques, collaborating with various teams to align software performance, analyzing and improving performance metrics, and implementing Metal device backend features for ML training acceleration technologies.

Job Details

Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating Machine learning libraries on Apple Silicon. Role has the opportunity to influence the design of compute and programming models in next generation GPU architectures. Responsibilities: Work on cutting-edge ML inference framework project and optimize code for efficient and scalable ML inference using distributed compute strategies such as data, tensor, pipeline and expert parallelism. Develop kernel and compiler level optimizations and perform in-depth analysis to ensure the best possible performance across Server hardware families. Apply advanced model optimization techniques including speculation, quantization, compression, and others to maximize throughput and minimize latency. Collaborate closely with hardware, compiler, and systems teams to align software performance with hardware capabilities. Analyze and improve performance metrics such as end-to-end latency, TTFT, TBOT, memory footprint, and compute efficiency. Implement features of Metal device backend for ML training acceleration technologies If this sounds of interest, we would love to hear from you!

Similar Jobs

bytedance - Senior Site Reliability Engineer, ML System - Foundation Model

bytedance

Seattle, Washington, United States (On-Site)
6 Months ago
CAE - Financial Analyst - Planning & Forecasting

CAE

Montreal, Quebec, Canada (On-Site)
1 Month ago
Marvell - Senior Manager, HR Communications

Marvell

Santa Clara, California, United States (On-Site)
1 Month ago
Guardian - Annuity Relationship Manager

Guardian

Bethlehem, Pennsylvania, United States (Hybrid)
2 Months ago
Rippling - Senior Manager, Implementation APAC

Rippling

Sydney, New South Wales, Australia (Hybrid)
1 Month ago
gismart - Machine Learning Engineer

gismart

(On-Site)
2 Weeks ago
Apple - Machine Learning Resource Management Engineer - SIML

Apple

Seattle, Washington, United States (On-Site)
1 Week ago
Hike - ML Engineer Intern

Hike

Delhi, India (Remote)
1 Week ago
Ubisoft - Lead R&D Scientist

Ubisoft

Shanghai, Shanghai, China (On-Site)
3 Months ago
Passive Logic - AI Control Theory & Optimization Scientist

Passive Logic

Salt Lake City, Utah, United States (On-Site)
7 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Nagarro - Associate Principal Engineer - Delivery Manager

Nagarro

Costa Rica, Mato Grosso Do Sul, Brazil (Remote)
9 Months ago
BetterMe - HR Generalist

BetterMe

Kyiv, Kyiv City, Ukraine (On-Site)
3 Weeks ago
Pattern - Senior CRM & Ecommerce Consultant

Pattern

Sydney, New South Wales, Australia (Hybrid)
1 Month ago
HP - Supply Chain Intern

HP

Barueri, São Paulo, Brazil (On-Site)
7 Months ago
Venoeer - Requirements Management Tool Support

Venoeer

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
Paytm - Messaging Solution - DM/Manager - Payments (MMA)

Paytm

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Dialpad AI - IT Support Specialist

Dialpad AI

Pasig, Metro Manila, Philippines (On-Site)
2 Months ago
Fieldguide - Principal Program Manager

Fieldguide

San Francisco, California, United States (Remote)
1 Week ago
USE Insider - Senior Software Engineer (Data Service Platform)

USE Insider

Istanbul, İstanbul, Türkiye (Remote)
11 Months ago
Vercel - Field Marketing Manager, EMEA

Vercel

London, England, United Kingdom (Remote)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Cupertino, California, United States

Apple - Machine Learning Engineer, GenAI & LLM, IS&T AiDP

Apple

Sunnyvale, California, United States (On-Site)
2 Months ago
Payactive - Director of Card Services

Payactive

Milpitas, California, United States (Hybrid)
9 Months ago
Springer Group - Senior Publisher, BMC Academic Journals

Springer Group

New York, United States (Hybrid)
2 Days ago
Grammarly - Principal, Strategy & Operations - M&A

Grammarly

San Francisco, California, United States (Hybrid)
1 Week ago
Nintendo - Technology Engineer (Retro Studios)

Nintendo

United States (Remote)
11 Months ago
CharacterAI - Lead Growth Product Manager

CharacterAI

San Francisco, California, United States (On-Site)
3 Months ago
Nice - Product Manager

Nice

United States (Hybrid)
2 Weeks ago
Dynamis Inc - Senior Scientific Advisor

Dynamis Inc

Huntsville, Alabama, United States (On-Site)
2 Weeks ago
Roblox - Senior Director of Engineering, User Connection

Roblox

San Mateo, California, United States (On-Site)
1 Month ago
Aerovect - Vehicle Service Technician, Autonomous Vehicles

Aerovect

Atlanta, Georgia, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Research Development Jobs

DevRev - Architect - Applied AI Engineer

DevRev

(Remote)
2 Months ago
bytedance - Research Scientist Graduates, Large Language Model (Doubao) - 2025 Start

bytedance

San Jose, California, United States (On-Site)
8 Months ago
bytedance - Student Researcher (Doubao (Seed) - Foundation Model AI Platform) - 2025 Start (PhD)

bytedance

San Jose, California, United States (On-Site)
8 Months ago
Apple - AIML - Senior Engineering Program Manager, Foundation Models

Apple

Cupertino, California, United States (On-Site)
1 Week ago
Riot Games - Research Operation Coordinator - Global Research Operations Team (Contract)

Riot Games

Shanghai, Shanghai, China (On-Site)
3 Months ago
Ansys - R&D Engineer II ( C++, GPU, Meshing)

Ansys

Pune, Maharashtra, India (On-Site)
2 Months ago
oportun - Senior Staff AI Software Engineer

oportun

Mexico (Remote)
1 Month ago
bytedance - Research Scientist Graduate (High-Performance Computing (Algorithm Acceleration)- Vision AI Platform)

bytedance

San Jose, California, United States (On-Site)
3 Months ago
Stacklok - Staff Software Engineer - Applied AI

Stacklok

Bellevue, Washington, United States (Hybrid)
2 Weeks ago

Get notifed when new similar jobs are uploaded

About The Company

Cupertino, California, United States (On-Site)

Cupertino, California, United States (On-Site)

Los Angeles, California, United States (On-Site)

Seattle, Washington, United States (On-Site)

Boulder, Colorado, United States (On-Site)

Cupertino, California, United States (On-Site)

Seattle, Washington, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Apple

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug