Staff Compute Infra Engineer - GPU & AI Systems

xAI; Palo Alto (CA); On-site; USD 180,000 - 440,000; Full time

Benefits offered by this job

Equity

Comprehensive medical, vision, and dental coverage

Access to a 401(k) retirement plan

Short & long-term disability insurance

Life insurance

Various discounts and perks

Job summary

xAI is seeking a talented individual to join their Compute Infrastructure team, focused on building one of the world’s largest AI supercomputers. You will design and optimize massive GPU clusters, ensuring fast and reliable AI training. Ideal candidates will possess deep programming skills, GPU kernel optimization experience, and a strong grasp of large-scale distributed systems.

This role offers a competitive salary range of $180,000 - $440,000, along with equity and comprehensive benefits including medical coverage and a 401(k) retirement plan.

Qualifications

• Deep low-level systems programming skills in C/C++ or Rust.
• Experience with large-scale distributed compute infrastructure.
• Hands-on experience with GPU kernel optimization techniques.

Responsibilities

• Design and build GPU clusters for extreme-scale training.
• Develop and optimize low-level CUDA kernels.
• Work on Linux kernel internals and resource management.

Skills

Deep low-level systems programming (C/C++ or Rust)

Experience building high performance exabyte scale storage systems

Strong experience with large-scale GPU clusters

Hands-on work with GPU kernel optimization

Experience with Linux kernel internals

Ability to reason from first principles

Job description

xAI is seeking a talented individual to join their Compute Infrastructure team, focused on building one of the world’s largest AI supercomputers. You will design and optimize massive GPU clusters, ensuring fast and reliable AI training. Ideal candidates will possess deep programming skills, GPU kernel optimization experience, and a strong grasp of large-scale distributed systems.

This role offers a competitive salary range of $180,000 - $440,000, along with equity and comprehensive benefits including medical coverage and a 401(k) retirement plan.