AMD

4 weeks ago

Sr. Software Development Engineer - Collectives and Network

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI

Last checked: 1 hour agoStatus: Live

Job Summary

Drive AMD’s AI pre-training and distributed inference performance by optimizing the Collectives and Network stack across hardware, AI frameworks, compilers, runtime, ROCm, and developer tools. Partner with hardware architecture, software optimization, and modeling teams to scale performance for multi-generational GPUs and state-of-the-art AI models. Lead performance analysis, tooling, and strategy for future hardware-software co-design.

Location: San Jose

Workplace: Hybrid

Employment Type: Full time

Job Function: Product Management

Key Responsibilities

•Help with 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.

Key Requirements

•Deep knowledge of Network, NIC and GPU hardware architecture, software optimization, performance modeling, and the latest trends in inference and training optimization.

Experience:AI hardwareGPUDistributed inferenceHigh performance computing

Education:PhD / Doctorate in Computer Science

Skills:PyTorchJAXVLLMSGLangPythonCUDAGPUNetworkingLoad balancingModel shardingPerformance profilingFrameworks

Languages:English

Tech Stack:PyTorchJAXVLLMSGLangPythonCUDAROCm

Company Brief

AMD

Designs and produces semiconductor products including CPUs, GPUs, and adaptive SoCs for consumer, enterprise, and embedded markets, competing across PCs, data centers, and gaming industries.

Industry: Electronics Manufacturing

Company Size: Enterprise (1,001+ employees)

Revenue: USD 1B+

Growth: Public Company

Valuation: Public Company (Market Cap in USD)

Funding: IPO / Publicly Listed

Headquarters: Santa Clara, United States

Founded: 1969

Glassdoor: 3.9

Website LinkedIn

Loading job details...

Preparing the role view and application actions.

AMD