Fellow, AI Performance Software Engineer
职位介绍
Line 1: You will work with a team to enable deep learning models, libraries, and applications for Instinct GPUs in on-prem and cloud environments.
Line 2: Analyze, profile, and optimize AI software to understand and overcome hardware bottlenecks, pushing performance toward roofline.
Line 3: Collaborate on Python and C++ components, ensuring integration with leading frameworks such as PyTorch or TensorFlow in inference, fine-tuning, and training.
Line 4: Experience with performance-driven software development and architecture understanding of GPUs is a plus; contribute to open-source development when possible.
Line 5: Work with profiling tools across the AI stack (TorchProfiler, ROCm profiler, VTune, Nsight) and GPU acceleration methods (NCCL/RCCL, OpenMP, MPI).
Line 6: Communicate progress clearly to leadership and stakeholders, and continuously learn tools and methods to improve software products.
Line 2: Analyze, profile, and optimize AI software to understand and overcome hardware bottlenecks, pushing performance toward roofline.
Line 3: Collaborate on Python and C++ components, ensuring integration with leading frameworks such as PyTorch or TensorFlow in inference, fine-tuning, and training.
Line 4: Experience with performance-driven software development and architecture understanding of GPUs is a plus; contribute to open-source development when possible.
Line 5: Work with profiling tools across the AI stack (TorchProfiler, ROCm profiler, VTune, Nsight) and GPU acceleration methods (NCCL/RCCL, OpenMP, MPI).
Line 6: Communicate progress clearly to leadership and stakeholders, and continuously learn tools and methods to improve software products.
查看完整职位
工作职责、任职要求、技能与福利 — 免费创建账号即可查看。
已有账户？ 登录
相似职位
其他可能适合您的职位。
? Applied Machine Learning Engineer, Circuit Design - New College Grad 2026 ? Applied Machine Learning Engineer - VLSI Design ? Machine Learning Engineer II, Sponsored Products and Brands ? Software Engineer- AI/ML, AWS Neuron Distributed Training ? Applied Scientist, LLM Code Agents, Kiro Science ? Senior Data Engineer, Ops Data Platform