Jobs / App***
On-Device ML Infrastructure Engineer (ML User Experience APIs), Graphics, Games and Machine Learning
App*** · Cupertino, CA, United States
Visa sponsorship details are locked. Unlock company name and apply link with .
Cupertino, CA, United States150,400-277,600 USD/yearlyOnsite
Remuneration
150,400-277,600 USD/yearly
Location
Cupertino, CA, United States
Visa sponsorship
Sponsors visa
Job summary
The On-Device ML Infrastructure Engineer role at App*** focuses on developing APIs for ML engineers to efficiently import and implement models on App*** platforms. The position involves integrating ML tools into repositories, optimizing model performance, and shaping the developer experience for ML development.
Benefits
Comprehensive medical and dental coverageRetirement benefitsDiscounted products and free servicesReimbursement for certain educational expenses including tuitionDiscretionary bonuses or commission paymentsRelocation assistance
Qualifications
- Experience with C++, Swift, and/or GPU programming paradigms
- Familiarity with QAT and other compression and quantization techniques employing PyTorch workflows
- Experience designing Python APIs and deploying production-grade Python packages
- Experience with MLIR/LLVM or similar compiler toolchains
- Familiarity with Hugging Face or other model repositories
Responsibilities
- Develop APIs in App***’s ML stack for ML engineers to efficiently import and implement their models
- Integrate App***’s ML tools into internal and external model repositories to demonstrate and stress-test model ingestion with peak efficiency and performance
- Develop optimizations across the pipeline, including source-level transformations, and custom operations to improve inference efficiency
- Onboard the latest ML models with peak performance, and use these examples to highlight and validate the authoring and runtime capabilities of App***’s inference stack
Skills
C++Python
Degrees
Bachelors in Computer Sciences, Engineering, or related subject area
Relocation
No