← All roles

Sr. Staff Software Development Engineer - Collectives and Network optimization

AMD· Engineering
Apply on company site
San Jose, CA$204k - $306kFull-timePosted 3mo ago

About the role

Senior level engineer responsible for strategy, architecture, optimization and tooling to achieve AI pre-training and distributed inference performance on AMD GPUs, focused on Collectives and Network performance across hardware, software, frameworks and runtimes.

Responsibilities

  • Help set strategy and roadmap for AMD Collectives and Network optimizations.
  • Provide guidelines to customers on efficient network load-balancing, workload scheduling and model sharding strategies.
  • Performance tuning, profiling and analysis of large-scale models for LLM, diffusion, multimodal, RecSys and generative AI, single node and distributed.
  • Participate in hardware-software co-design for future hardware optimizations – especially on scale-up networks, NIC and scale-out networks.
  • Develop and improve framework, tools and infrastructure for performance estimation, modeling and reporting.
  • Communicate and present the results of the performance analysis and modeling to stakeholders, and senior leadership. And provide a concrete recommendation.
  • Cross team collaboration and working across the organization to identify opportunities and develop strategies.

Qualifications

  • A PhD or master's degree in computer science, electrical engineering, or a related field.
  • Multiple years of technical experience in performance optimization.
  • Strong technical expertise and experience in performance analysis, projection, and network hardware architecture.
  • Deep knowledge and hand-on experience of AI Frameworks such as PyTorch, JAX, vLLM, and SGLang.
  • Strong technical leadership skills, ability to work collaboratively with cross-functional teams.
  • Mentor, coach, and inspire a diverse and talented team of researchers and engineers.
  • Excellent written, verbal, and presentation skills, ability to coordinate internally and externally.

Skills

Required:Network optimizationPerformance profiling and analysisAI frameworks
Nice to have:Hardware-software co-design

Benefits

  • AMD benefits at a glance
  • Not explicitly listed in detail in the description
Last updated Oct 15, 2026. Interested? Apply on the company site.
AMD logo

About AMD

together we advance_AI

AMD delivers leadership high-performance and adaptive computing solutions to advance data center AI, AI PCs, intelligent edge devices, gaming, & beyond.

AIData CenterHigh-Performance ComputingGamingEdge Devices
Company profileWebsite

More open roles at AMD