מציאת משרת הייטק בחברות הטובות ביותר מעולם לא הייתה קלה יותר
Apple Model Optimization Engineer PyTorch Infrastructure Development United States, California, Cupertino 162832441
08.06.2025
In this role, the Model Optimization Engineer will be an expert in understanding the internal workings of PyTorch, graph capturing and graph editing mechanisms, methods to observe and modify intermediate activations and weights, tensor subclasses, custom ops, different types of parallelism for training models, and use this knowledge to implement and update the core infrastructure of the optimization library which enables an efficient and scalable implementation of various classes of compression algorithms. You'll also set up and debug training jobs, datasets, evaluation, performance benchmarking pipelines. Additionally, you will...- Design and develop the core infrastructure which powers the implementations of various compression algorithms (training time, post training, data free, calibration data based etc)- Implement the latest algorithms from research papers for model compression in the optimization library.- Design clean, intuitive, maintainable APIs - Run detailed experiments and ablation studies to profile algorithms on various models and tasks, across different model sizes.
- Bachelors in Computer Sciences, Engineering, or related discipline.
- 3+ years of industry and/or research experience
- Highly proficient in Python programming
- Proficiency in at least one ML authoring framework, such as PyTorch, TensorFlow, JAX, MLX
- Experience in the area of model compression and quantization techniques, specially in one of the optimization libraries for an ML framework (e.g. torch.ao).
- Ability to ramp up quickly on new training code bases and run experiments.
- Demonstrated ability to design user friendly and maintainable APIs
- Experience in training, fine tuning, and optimizing neural network models
- Primary contributor to a model optimization/compression library.
- Self prioritize and adjust to changing priorities and asks
- Improving model optimization documentation, writing tutorials and guides
- Good communication skills, including ability to communicate with cross-functional audiences
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
משרות נוספות שיכולות לעניין אותך
