Jobs in the United States

25+ jobs

Save this search

Added 13 hours ago
O
Lead Principal Systems Software Engineer
Oracle
United States
- $146,300 to $306,400 USD / year
Be the senior technical engineer and hands-on architect for the Cloud Infrastructure and Services platform, spanning physical provisioning, virtualization, Kubernetes and platform services.…
Added 17 hours ago
Senior Lead Software Engineer- AI Platform engineer
JPMC page
Seattle, WA · Palo Alto, CA · San Francisco, CA
As a Senior Lead Software Engineer at JPMorgan Chase within the Corporate Sector, Infrastructure Platforms team, you are an integral part of an agile team that works to enhance, build, and…
Added 18 hours ago
Senior Backend Software Engineer
Flexcompute Inc.
Madison, WI
We are looking for a Senior Backend Software Engineer to join our team in Madison, Wisconsin, and help us design and implement scalable solutions that meet the growing demands of our…
Added 2 days ago
Senior Deep Learning Systems Architect
Nvidia
Santa Clara, CA
We are now looking for a Senior Deep Learning Systems Architect!
Added 3 days ago
Senior AI Engineer, AI Safety
NVIDIA USA
Remote, United States
Strong engineering fundamentals include deep proficiency in one or more languages such as Rust or Python. Experience with machine learning frameworks like PyTorch is also required. The…
Added 3 days ago
Senior AI Software Developer & Engineer (US Citizen)
Remote, United States
Deep expertise in machine learning and deep learning, particularly transformer-based models and LLMs.
Added 3 days ago
Distinguished Signal and Power Integrity Engineer
NVIDIA USA
Remote, United States
…continuously reinvented itself over two decades. Our invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel…
Added 4 days ago
Cloud Solution Engineer (Genomics & AI)
Oracle
Jefferson City, MO
…modeling. Assess genomic foundation models and sequence-based deep learning where they address a defined scientific…
Added 4 days ago
Senior Backend Software Engineer
Flexcompute Inc.
Madison, WI
We are looking for a Senior Backend Software Engineer to join our team in Madison, Wisconsin, and help us design and implement scalable solutions that meet the growing demands of our…
Added 4 days ago
Principal Robotics Platform Architect
Irving, TX
The Principal Digital Architect for the Robotics Platform is responsible for defining and delivering the end-to-end software architecture of Caterpillar’s robotics platform - the…
Added 4 days ago
Principal Site Reliability Engineer
NVIDIA USA
CA
…It combines software and systems engineering practices with deep expertise across distributed systems, networking, cloud infrastructure, Kubernetes, databases, capacity management,…
Added 4 days ago
Distinguished Engineer, AI Security Systems (Remote)
CrowdStrike
Austin, TX
As one of the Lab’s most senior technical leaders, you will architect systems that turn frontier AI into autonomous, trustworthy cyber defense - from the sensor to the cloud. You will write…
Added 4 days ago
Lead Solution Architect, Physical AI
World Wide Technology
Maryland Heights, MO
The Lead Solution Architect, Physical AI is the technical owner of WWT's Physical AI engagements, accountable for the design and delivery of systems that perceive the physical world through…
Added 4 days ago
Staff AI Architect
Domino's
Ann Arbor, MI
As Staff AI Architect, you own the technical architecture for Domino's enterprise AI platform – spanning classical AI, generative AI, and agentic AI. This is the senior…
Added 4 days ago
Senior Solutions Architect, DL Training
NVIDIA USA
WA · CA
NVIDIA is building the world’s leading AI company, and we are looking for an experienced Senior Solutions Architect to help customers build, scale, and optimize AI training solutions using…
Added 5 days ago
Senior Data Engineer
Marble Technologies
Lincoln, NE · Omaha, NE · Cambridge, MA
As a Senior Data Engineer at Marble, you will own the design and performance of data pipelines that power everything from real-time classification dashboards to ML training datasets to…
Added 5 days ago
Senior/Principal Computer Scientist - Space, Satellites, and Remote Sensing Systems, Onsite
Sandia National Laboratories
Albuquerque, NM
…extraction, and related applications using neural networks, deep learning, or transformer-based…
Added 5 days ago
Z
Staff / Senior Staff Machine Learning Engineer - Collision Avoidance System
Zoox
Foster City, CA
- $291,000 to $409,000 USD / year
As a Staff or Senior Staff engineer, you will be the technical lead for CAS algorithms, with a focus on CAS perception. You will shape how CAS sees the world, combining interpretable…
Added 6 days ago
Senior Staff/Principal Engineer, Foundation Model Compute Infra
Apple
Cupertino, CA · Santa Clara, CA · Seattle, WA
As a Senior Staff/Principal Engineer on the Foundation Model Compute Infrastructure team, you will architect, build, and operate large-scale infrastructure powering foundation model…
Added 6 days ago
Data Science Analyst Senior/Intermediate
University of Michigan
Ann Arbor, MI
In this position, you will work under the general direction of the MAISE AIG Manager. As a Data Architect Intermediate/Senior, you will design, develop, validate, deploy, and maintain data…
Added 7 days ago
L
Senior Security Engineer - Detection & Response
Lambda
San Francisco, CA · San Jose, CA · Bellevue, WA
- $296,000 to $395,000 USD / year
…give everyone the power of superintelligence. One person, one…
Added 7 days ago
Field CTO, Financial Services AI Infrastructure
Nebius
United States
The Field CTO for Financial Services is a senior technical leadership role at the intersection of production GPU infrastructure and the regulated business of financial services. You will…
Added 7 days ago
Senior Software Engineer (JPE - 056)
SageCor Solutions
Fort Meade, MD
SageCor is seeking a senior software engineer to design and build high-performance acceleration pipelines for a government mission program. The work centers on GPU architecture and…
Added 8 days ago
Senior Silicon Security Architect
NVIDIA Corporation
North District, IL
NVIDIA’s invention of the GPU 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning…
Added 8 days ago
Senior Software Architect, Advanced Development
NVIDIA Corporation
North District, IL
As a Senior architect in the Advanced Development team, you will explore technological challenges on accelerate networking and building AI data centers. Research new transport functions and…

Advanced Micro Devices, Inc

Principal Distributed Systems Software Engineer, Collectives

San Jose, CA, United States

Apply About employer

Applications open in a new tab.

Job details

Apply by: Oct 28, 2027
Posting language: English
Posted: Oct 28, 2026

Original job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD. At AMD, we believe technology has the power to solve the world’s most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMD is shaping the future.

Whether you’re designing next-gen processors, enabling AI breakthroughs, or bringing leading edge products to market, every role at AMD contributes to something bigger - technology that moves the world forward. Join us and, together, we’ll advance your career.

THE ROLE:

As a core member of the Collectives and Network Performance team, you will play a pivotal role in defining and delivering industry-leading AI pre-training and distributed inference performance on AMD GPUs. Your expertise will be critical in optimizing collective communication libraries, scale-up and scale-out networking software, and training/inference performance across multi-GPU and multi-node systems. You will engage with internal hardware, GPU library, compiler, runtime and ROCm teams as well as open-source maintainers to ensure seamless integration of optimizations, applying advanced engineering principles to drive continuous improvement across multiple generations of AMD GPUs and a wide range of state-of-the-art AI models.

THE PERSON:

Seeking an industry-leading expert in collective communication and network performance, with advanced technical and analytical skills in Linux environments and deep C++ expertise. The ideal candidate will excel in providing technical leadership, guiding teams, and driving projects and initiatives independently. You will define goals, scope, and own development efforts while collaborating effectively within a high-performing team. You have deep knowledge of network, NIC and GPU hardware architecture, hands-on experience mapping model architectures to low-level software and hardware, and a strong grasp of current trends in distributed training and inference at scale.

KEY RESPONSIBILITIES:

Drive Collectives & Network Strategy: Define goals, scope and roadmap for a major area of AMD’s collectives and network optimizations, and align them with hardware, framework and ROCm roadmaps.
Architect Collective Communication Libraries: Lead the design and development of collective algorithms and optimizations (e.g., all-reduce, all-to-all, all-gather) across scale-up and scale-out networks.
Optimize Distributed Training & Inference: Lead performance tuning, profiling and analysis of large-scale LLM, diffusion, multimodal, RecSys and generative AI models on single-node and multi-node systems, and drive the tradeoffs and design decisions.
Hardware-Software Co-Design: Participate in co-design of future scale-up networks, NICs and scale-out networks, and represent software performance requirements to hardware teams.
Performance Modeling & Tooling: Drive frameworks, tools and infrastructure for performance estimation, modeling, regression tracking and reporting.
Collaborate with Framework & Library Teams: Work closely with internal GPU library, compiler and runtime teams, and with open-source maintainers (e.g., PyTorch, JAX, vLLM, SGLang), to ensure optimizations are aligned with requirements and integrated upstream.
Customer & Partner Guidance: Provide guidelines on efficient network load balancing, workload scheduling and model sharding strategies.
Communicate with Leadership: Present performance analysis and modeling results to stakeholders and senior leadership, with concrete recommendations.
Software Engineering Best Practices: Apply sound engineering principles and set standards to ensure robust, maintainable solutions.
Lead, Guide & Mentor: Provide strategic direction and mentorship to junior team members, fostering growth and collaboration through code reviews, knowledge sharing and technical guidance.

PREFERRED EXPERIENCE:

Collective Communication & Network Expertise: Deep expertise in designing and optimizing collective communication libraries (e.g., RCCL, NCCL, MPI) and a thorough understanding of NIC, RDMA, scale-up and scale-out network architectures and their impact on model performance.
Performance Analysis & Modeling: Strong technical expertise and proven experience in performance analysis, projection and bottleneck identification across the full stack, from model to hardware.
Deep Learning Integration: Proven ability to integrate optimized communication into ML frameworks such as PyTorch, JAX, vLLM and SGLang, with a focus on throughput, scalability and efficient execution for training and inference.
Software Engineering Excellence: Advanced proficiency in C++ and Python with deep experience in performance tuning, debugging and robust test design, ensuring reliable, maintainable, high-performance codebases.
High-Performance Computing: Broad and in-depth experience with large-scale, heterogeneous compute environments; adept at optimizing AI workloads for performance, efficiency and resource utilization across clusters.
Technical Leadership & Communication: Demonstrated ability to lead initiatives across teams, mentor and coach engineers, and communicate effectively in writing and presentations, both internally and externally.
Professional experience: Senior level of industry experience in technical software development, with a focus on performance engineering, communication libraries, networking, and GPU/distributed systems.

ACADEMIC CREDENTIALS:

Master’s and/or PhD degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.

LOCATION:

San Jose, CA or Austin, TX. Onsite.

This role is not eligible for visa sponsorship.

Benefits offered are described: AMD benefits at a glance.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s “Responsible AI Policy” is available here.

This posting is for an existing vacancy.

Apply About employer

Report this job