Senior Software Engineer, AI Triton Communication

Amd

Confirmed live yesterday High trust
Hybrid

Quick summary

Work type
Hybrid
Location
San Jose, CA
Salary
$164,000–$246,000 / yr
Posted
7 days ago
Freshness
Confirmed live yesterday
Closes
Sep 4, 2027

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $185k
This role $205k
$139k most similar roles pay here $257k

This role pays more than 68% of similar roles. Most pay $154,900–$215,625 — the shaded band above. At the midpoint, this role pays about $205k versus about $185k for comparable roles.

Based on 240 similar postings.

Employer

About Amd

AMD (Advanced Micro Devices) is a semiconductor company that develops high-performance processors, graphics cards, and adaptive computing solutions for gaming, data centers, and embedded markets. Industry: Semiconductors

Amd currently has 367 open roles on FindRole.

Listed pay typically runs $166,400–$249,600 across 367 roles with salary data.

Most-posted roles

View all roles at Amd

At a glance

TL;DR · Senior Software Engineer, AI Triton Communication

Sr. Software Engineer - AI Triton Communication will join the team to advance the Triton compiler and runtime stack for AMD CDNA and next-generation GPU architectures. The role focuses on building native distributed execution and communication capabilities, including collective operations, remote memory access, and synchronization primitives to support large-scale training and inference. You will develop infrastructure to optimize inter-GPU data movement, computation overlap, and hardware utilization while collaborating with architecture and software teams. Key technical requirements include expertise in GPU architecture, compiler technologies, and distributed systems. Candidates should possess skills in Triton, ROCm, HIP, CUDA, MLIR, or LLVM. Experience with communication libraries like RCCL, NCCL, NVSHMEM, rocSHMEM, or MPI is preferred. The work addresses the critical challenge of scaling AI workloads across multiple GPUs by improving performance and scalability within the Triton ecosystem for high-performance computing.

What does a Software Engineer earn in California?

Median $214000 from 775 postings across 63 companies.

See salary data

What you'll do

  • Develop native distributed communication and execution capabilities within the Triton AMDGPU backend.
  • Implement compiler and runtime mechanisms for GPU-initiated communication, including collective operations and remote memory access.
  • Optimize performance across compute and communication by improving inter-GPU data movement and memory hierarchy utilization.
  • Create and optimize distributed Triton kernels to ensure high scalability for large-scale AI workloads.
  • Profile and debug complex cross-stack issues spanning the Triton compiler, ROCm stack, and GPU hardware.
  • Contribute to the open-source Triton and ROCm distributed ecosystems to drive innovation in GPU computing.

What we're looking for

  • Bachelor's or Master's Degree in Computer Engineering, Computer Science, Electrical Engineering, or equivalent practical experience.
  • Deep expertise in GPU architecture, compiler technologies, and distributed GPU systems.
  • Experience with Triton compiler and runtime.
  • Strong knowledge of MLIR and/or LLVM internals.
  • Proficiency in GPU programming using Triton, HIP, CUDA, or similar parallel programming environments.
  • Familiarity with ROCm, HIP, CUDA, or similar GPU programming ecosystems and performance profiling tools.
  • Knowledge of multi-GPU interconnects such as XGMI, NVLink, PCIe, or InfiniBand.
  • Experience with distributed GPU communication libraries like RCCL, NCCL, NVSHMEM, rocSHMEM, or MPI.

More like this

Similar roles

Senior Deep Learning Frameworks CUDA Software Engineer

Nvidia

Remote (Santa Clara, CA) +1 11 days ago $184,000$287,500
CUDA PyTorch JAX C++ Python TRT-LLM vLLM SGLang TensorRT Triton NCCL MPI UCX XLA HPC Kernel Authoring NVIDIA Nsight Systems Compiler Technologies
8+ yrs exp Remote

Senior Software Engineer, Siri Speech

Apple Inc

Cupertino, CA 91 days ago $184,700$324,800
Swift Objective-C Machine Learning Generative AI Speech-to-Text API Design Software Architecture iOS iPadOS macOS watchOS visionOS

Senior Software Engineer

Adobe

San Jose, CA 179 days ago $177,900$257,550
Python Spark SQL Apache Airflow Databricks Snowflake DAX Tableau Power BI Adobe Analytics ETL Machine Learning Statistical Modeling Predictive Analytics A/B Testing Distributed Computing
3+ yrs exp

Senior Software Engineer

Alteryx

Remote 140 days ago $119,000$153,000
Python C++ gRPC CI/CD Git NumPy CPython SQLite OpenSSL WiX REST) API Node.js React JavaScript JSON Docker Kubernetes Rust AWS GCP Azure
Remote

Senior Software Engineer

Palo Alto Networks

Santa Clara, CA 91 days ago $106,000$170,500
Python Node.js TypeScript SQL MongoDB BigQuery GCP AWS Azure CI/CD REST APIs Kubernetes Docker AI/ML Pipelines Data Science Model Context Protocol
2+ yrs exp