The point where experts and best companies meet
Apple AIML - Software Engineer United States, California, Cupertino 620091002
31.08.2024
Share
Description
Key Responsibilities - Evaluate and integrate hardware accelerators for the training and inference of deep learning models. - Integrate new technologies (eg. torch.compile, XLA, Model Parallelism, Tensor Parallelism, FP8) into the existing codebase to enable large-scale deployment of ML models, significantly reducing engineering iteration time. - Profile system performance, identify bottlenecks, and optimize runtime performance. - Work closely with external partners to ensure smooth integration of hardware accelerators with our existing infrastructure. - Stay up-to-date with the latest advancements in hardware accelerators and deep learning technologies.
Minimum Qualifications
- Excellent Python coding skills
- Experience with GPUs, TPUs, AWS Trainium
- Familiarity with PyTorch
- Excellent interpersonal skills
Preferred Qualifications
- ML System experience - Large Scale distributed Training and Inference.
- Familiarity with JAX, XLA, HLO, Torch Compile
- Experience implementing/optimizing CUDA kernels
Pay & Benefits
- At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $121,909 and $214,500 annualized, and your base pay will depend on your skills, qualifications, experience, and location.Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics.
These jobs might be a good fit
