Software Engineer - Infrastructure

xAI; Palo Alto (CA); On-site; USD 125,000 - 189,000; Full time

Job summary

A leading technology company is seeking a Software Engineer for their Infrastructure team. This role includes designing recommendation algorithms, working with advanced data infrastructures, and contributing to an impactful project that enhances user experience on a platform with millions of users. Candidates should possess strong skills in data handling and deep learning frameworks, with opportunities for professional growth within a collaborative and innovative environment.

Qualifications

• Knowledge of data infrastructure like Kafka, Clickhouse, and Spark required.
• Experience with recommender systems and deep learning applications preferred.
• Exceptional candidates may have experience writing CUDA kernels.

Responsibilities

• Design and architect recommendation algorithms across various product surfaces.
• Write data pipelines and training jobs that learn from product data.
• Iterate and improve algorithms through real-time user feedback and experimentation.

Skills

Data Infrastructure

Deep Learning

Recommendation Systems

Docker

Python

Tools

Kafka

Clickhouse

Spark

JAX

PyTorch

CUDA

Job description

Join to apply for the Software Engineer - Infrastructure role at xAI

2 days ago Be among the first 25 applicants

Join to apply for the Software Engineer - Infrastructure role at xAI

xAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. The team is small, highly motivated, and focused on engineering excellence with a flat organizational structure. The Applied AI Engineer role involves working on a high-priority project integrating xAI’s expertise into the X platform, which has approximately 600 million monthly users. Responsibilities include designing and architecting recommendation algorithms across various product surfaces in X (Timelines, Ads, Video, Search, Notifications), leveraging xAI’s infrastructure and AI stacks to enhance user experience, writing data pipelines and training jobs that learn from product data, iterating and improving algorithms through real-time user feedback and experimentation, and ensuring scalability and efficiency of machine learning systems. Candidates should have knowledge of data infrastructure like Kafka, Clickhouse, and Spark, experience with recommender systems and/or deep learning applications at industrial scale, and skills in DL frameworks such as JAX or PyTorch. Exceptional candidates may have experience writing CUDA kernels. The interview process includes a phone interview, a deep dive coding challenge, and a meet and greet with the wider team. The role is based in Palo Alto, CA, and candidates are expected to be located near the Bay Area or open to relocation.

Seniority level

• Seniority level

  Entry level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Engineering and Information Technology

• Industries

  Technology, Information and Internet

Referrals increase your chances of interviewing at xAI by 2x

Sign in to set job alerts for “Software Engineer” roles.

Software Engineer, AI Intern (Summer 2025)

Mountain View, CA $125,400.00-$188,100.00 2 weeks ago

Software Engineer, AI Platform - New Grad

Menlo Park, CA $56.25-$173,000.00 2 weeks ago

Alameda, CA $130,000.00-$160,000.00 5 days ago

(General Hire) Software Engineer Graduate (Advertisement Team) - 2025 Start (BS/MS)

San Jose, CA $113,500.00-$250,000.00 2 weeks ago

Software Engineer 4 - TV & Web Player Platform

Sunnyvale, CA $167,000.00-$185,500.00 5 days ago

Full Stack Software Engineer - Post-training

Full Stack Software Engineer (L4), Product Localization Engineering

Software Engineer(s) - New Grad (Fall 2025 Graduation)

San Jose, CA $133,900.00-$242,000.00 4 days ago

New College Grad Software Engineer, Software Engineering Development (Apps)

San Jose, CA $92,735.00-$131,300.00 1 week ago

San Jose, CA $113,400.00-$206,300.00 2 weeks ago

eCommerce Full Stack Developer (React / Shopify) - On Site

Sunnyvale, CA $117,000.00-$234,000.00 2 days ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.