On-device ML Infrastructure Engineer (ML Performance Visualizatio...
Apple Inc. - Seattle, Washington, us, 98127
Work at Apple Inc.
Overview
- View job
Overview
Seattle, Washington, United States | Software and Services Description
We are building the first end-to-end developer experience for ML development that leverages Apple’s vertical integration. This allows developers to iterate on model authoring, optimization, transformation, execution, debugging, profiling, and analysis. This role offers a great opportunity to scale and extend a significant on-device ML benchmarking service used across Apple, supporting a range of devices from wearables to Apple Silicon Macs. The role involves building an end-to-end developer experience for ML development, exploring on-device machine learning research areas such as model visualization, efficient inference algorithms, model compression, and ML compilers/run-time. Key responsibilities: Drive UI/front-end experiences for a ML benchmarking service in a fast-paced environment Explore intelligent visualization and insights of on-device ML models Maintain the health and performance of the ML benchmarking service, including debugging failures and addressing user questions Collaborate with ML and hardware teams across Apple Minimum Qualifications
Strong programming and software design skills in Python Knowledge of ML fundamentals including training, evaluation, and deployment/inference Passion for ML, especially on-device ML use cases Excellent collaboration and communication skills Preferred Qualifications
Masters or PhDs in Computer Science or relevant disciplines Experience with ML frameworks like PyTorch, TensorFlow, JAX, especially on-device frameworks such as CoreML, TFLite, or ExecuTorch Back-end skills including containers (Docker), cloud orchestration (Kubernetes), databases (SQL, Postgres) Experience with ML architectures such as Transformers, CNNs, or Stable Diffusion At Apple, base pay ranges from $135,400 to $250,600, depending on skills and experience. Employees may participate in stock programs, receive benefits like medical coverage, retirement plans, educational reimbursements, and may be eligible for bonuses or relocation assistance. Apple is an equal opportunity employer committed to diversity and inclusion. We promote equal opportunity regardless of race, color, religion, sex, sexual orientation, gender identity, national origin, disability, veteran status, or other protected characteristics.
#J-18808-Ljbffr