On-device ML Infrastructure Engineer (ML Performance Insights)
On-device ML Infrastructure Engineer (ML Performance Insights)
Posted on :
23-08-2025
Employer Active
1 Vacancy
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Share
Send me jobs like this
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Job Description
This role provides a great opportunity to help scale and extend an on-device ML benchmarking service that is used across Apple in support of a range of devices from small wearables up to the largest Apple Silicon this role you will be an integral member of a talented team that is building the first end-to-end developer experience for ML development that by taking advantage of Apples vertical integration allows developers to iterate on model authoring optimization transformation execution debugging profiling and analysis. The role further offers a learning platform to dig into the latest research about on-device machine learning an exciting ML frontier! Possible example areas include model visualization efficient inference algorithms model compression on-device fine-tuning federated learning and/or ML compilers/run-time.
- Strong ML fundamentals across training evaluation and inference and knowledge of modern model architectures such as Transformers CNNs or Stable Diffusion;
- Programming and software design skills (proficiency in Python and/or C/C);
- A passion for edge / on-device ML;
- Understanding about performance modeling analysis and profiling of computer systems and how to optimize code run time and throughput for a given platform;
- Collaboration product-focus and excellent interpersonal skills.
- Masters or PhDs in Computer Science or relevant disciplines;
- On-device ML frameworks such as CoreML TFLite or ExecuTorch;
- Experience with any ML authoring framework (PyTorch TensorFlow JAX etc.) is a strong plus;
- Experience in software architecture APIs high performance extensible software and scalable software systems;
- Understanding of how to optimize code run time and throughput for a given platform;
- Interest and experience in power and/or hardware accelerators is a plus;
- Back-end system skills including containers (docker) cloud orchestration (Kubernetes) database (SQL Postgres).
Employment Type
Full-Time
Key Skills
Report This Job
Disclaimer: Drjobpro.com is only a platform that connects job seekers and employers. Applicants are advised to conduct their own independent research into the credentials of the prospective employer.We always make certain that our clients do not endorse any request for money payments, thus we advise against sharing any personal or bank-related information with any third party. If you suspect fraud or malpractice, please contact us via contact us page.
