Proven track record of designing and scaling robust ML infrastructure and frameworks that support both training and inference across teams and orgs.
Experience of model quantization, tensor parallelism, and inference optimizations (e.g ONNX Runtime, TensorRT, vLLM). Actively led evaluation and adoption of such technologies.
Recognized as a technical leader and mentor, supports the growth of engineers through code/design reviews, working groups, and internal knowledge sharing.
Experience building machine learning models using frameworks like PyTorch, TensorFlow. Provides technical guidance and mentorship on best practices.
Prior experience in advertising industry, federated learning and privacy-preserving ML techniques.
Led development of foundational AI/ML platforms and tooling including Feature Stores, Vector DB to accelerate team productivity and model lifecycle management.
Experience working on distributed systems (e.g Ray, Spark, Kubernetes).
Experience performance tuning & trouble-shooting.
Pride in building tools to automate routine tasks, organized & detailed.
Passionate about developer experience builds abstractions, automation tools, and reusable components to streamline ML workflows and reduce operational burden.
Ability to communicate effectively, both written and verbal, with technical and non-technical multi-functional teams.
Results oriented with a desire to work in a fast-paced and collaborative work environment.
PhD/MS/BS in computer science or related field with 8+ years of experience in machine learning and strong software engineering skills.
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.