Fellow, AI Performance Software Engineer
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
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.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
Already have an account? Log in
Similar openings
Other roles that could suit you.
? 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