Finding the best job has never been easier
Apple Senior Machine Learning Engineer United States, New York, New York 480902303
Today
Share
- Experience designing and developing ML infrastructure/frameworks for training and inference.
- Experience of model quantization, tensor parallelism, and inference optimizations (e.g ONNX Runtime, TensorRT, vLLM).
- Experience building machine learning models using frameworks like PyTorch, TensorFlow.
- Experience building AI/ML tooling and/or infrastructure (e.g FeatureStore, VectorDB).
- 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.
- Familiarity with CI/CD tooling.
- Strong problem solving and debugging skills.
- PhD in Computer Science with 2+ years building production machine learning systems, or (b) MS in CS with 4+ years of engineering experience and 2+ years building production machine learning systems, or (c) BS in CS with 5+ years of engineering experience and 2+ years building production machine learning system.
- 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.
- Prior experience in advertising industry, federated learning and privacy-preserving ML techniques is a huge plus.
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
These jobs might be a good fit
