Network Engineer - AI/HPC

Xai

Memphis (TN)

On-site

USD 180,000 - 240,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Job summary

xAI is seeking an experienced engineer to design and operate large-scale ethernet AI/HPC networks. The role focuses on RoCEv2, NCCL usage, and building scalable GPU infrastructure for training and inference workloads. You will develop dashboards, optimize configurations, and own reliability across the fleet.

The position requires deep networking expertise, strong Python automation skills, and willingness to travel to Memphis for capacity expansion and on-call duties as the team grows.

Qualifications

  • Designing and operating large scale networks in AI/HPC environments (10+ years).
  • Expertise in Ethernet with Infiniband considered a bonus.
  • Ability to train and troubleshoot AI training/inference workloads and NCCL usage.
  • Develop a portfolio of metrics for performance and operations to optimize fleet.
  • Automate tasks and analyze large datasets with Python.

Responsibilities

  • Design and operate large scale network infrastructure for AI training/inference workloads.
  • Build and tune metric dashboards to optimize fleet performance and availability.
  • Design backend/frontend networks to enable scalable GPU infrastructure with minimal tooling.
  • Travel to Memphis for capacity expansion and participate in on-call rotation; contribute to deployment/ops frameworks.

Skills

Large‑scale networks
Ethernet AI HPC
NCCL debugging
Performance metrics
Python automation
Network optimization

Job description

ABOUT xAI

xAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for individuals who appreciate challenging themselves and thrive on curiosity. We operate with a flat organizational structure. All employees are expected to be hands-on and to contribute directly to the company’s mission. Leadership is given to those who show initiative and consistently deliver excellence. Work ethic and strong prioritization skills are important. All employees are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.

ABOUT THE ROLE:

xAI was first in the world to build a 100k GPU cluster on an ethernet network and then did it again in 92 days, floors, walls and all. We need an engineer with deep experience in RoCEv2 that can develop at hyper scale while optimizing performance and availability.

xAI is building at a furious pace with the latest hardware to help people understand the universe. To make the next significant leap forward, we need to own our own destiny by understanding our current network performance and availability and then optimize it to our training models and how we execute customer inference queries. You will spend most of your days deep inside NCCL, building metric dashboards and tweaking configurations to ensure no performance is left on the table. You will help design the next iteration of our backend and front-end networks that will allow us to seamlessly build-out new GPU infrastructure with little to no engineering assistance.

There will be a significant amount of travel to Memphis for building more capacity as well as participating in a team on-call rotation and helping on other scaling and maintenance efforts. This will become easier as we build out the team and engineers contribute to deployment and operations frameworks to remove repetitive tasks.

BASIC QUALIFICATIONS:
  • A minimum of 10 years designing and operating large scale networks with 5 years in the ethernet AI/HPC space.
  • Deep understanding of congestion control on ethernet with Infiniband an added bonus.
  • Deep understanding of AI training and inference workloads and how they operate on the network. As part of this you are able to use and debug NCCL and potentially commit to the library.
  • Expertise in creating a portfolio of metrics for performance and operations to optimize the fleet for training and inference traffic.
  • Experience with Python to automate away repetitive tasks and facilitate your daily job working with and analyzing large sets of data.

xAI is an equal opportunity employer. For details on data processing, view our Recruitment Privacy Notice.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Senior AI Networking Engineer - RoCEv2, NCCL, HPC
Senior AI Networking Engineer - RoCEv2, NCCL, HPC

Xai • Memphis (TN)

On-site
USD 180,000 - 240,000
Software Engineer - Networking Software and Services
Software Engineer - Networking Software and Services

Pantera Capital • Palo Alto (CA)

Hybrid
USD 180,000 - 440,000
Equity
Comprehensive medical, vision, and dental coverage
401(k) retirement plan
+2
Network Engineer
Network Engineer

Cerebras Systems, Inc. • Sunnyvale (CA)

On-site
USD 120,000 - 180,000
Autonomy in architecture decisions
Work culture that respects individual beliefs
Opportunities for continuous learning
Network Engineer
Network Engineer

Cerebras • Sunnyvale (CA)

On-site
USD 130,000 - 180,000
GPU Network Engineer
GPU Network Engineer

Blue Signal Search • Santa Clara (CA)

On-site
USD <240,000
Software Engineer - Networking Software and Services
Software Engineer - Networking Software and Services

xAI • Palo Alto (CA)

On-site
USD 150,000 - 250,000
Equity
Comprehensive medical, vision, and dental coverage
401(k) retirement plan
+3
Senior AI Network Engineer - AI Infrastructure
Senior AI Network Engineer - AI Infrastructure

Hamilton Barnes Associates Limited • United States

On-site
USD 220,000 - 350,000
Annual bonus
Equity opportunities
Flexible working arrangements
+1
Senior Software Engineer – AI Middleware
Senior Software Engineer – AI Middleware

Cornelis Networks, Inc. • Austin (TX)

Remote
Medical, dental, and vision coverage
401(k) with company match
Generous paid holidays
Network Development Engineer
Network Development Engineer

xAI • Memphis (TN)

On-site
USD 65,000 - 85,000
Network Operations Engineer, AI Networking
Network Operations Engineer, AI Networking

OpenAI • San Francisco (CA)

On-site
USD 140,000 - 210,000