July 14, 2026

Software Engineer - Training/Inference (C++)

Senior • On-site

129,996 - 234,996 USD/yr

Palo Alto, CA

SpaceXAI'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

We are building the high-performance inference platform that serves Grok to millions of users every day with lightning speed and perfect reliability.

As a Member of Technical Staff - Inference, you will design and optimize large-scale model serving systems end-to-end. You will own everything from distributed infrastructure (global KV cache, continuous batching, load balancing, auto-scaling) to deep low-level optimizations (GPU kernels, quantization, speculative decoding, tail latency).

This is a high-impact role where your work directly determines how fast and reliably users interact with Grok at massive scale.

Responsibilities

  • Architect and implement scalable distributed infrastructure for model serving (load balancing, auto-scaling, batch scheduling, global KV cache).
  • Optimize latency and throughput of model inference under real production workloads.
  • Build reliable, high-concurrency serving systems that serve billions of users with 100% uptime, 0% error rate, and excellent tail latency.
  • Benchmark, fine-tune, and accelerate inference engines (including low-level GPU kernel work and code generation).
  • Develop custom tools to trace, replay, and fix issues across the full stack — from orchestration down to GPU kernels.
  • Create robust CI/CD infrastructure for seamless endpoint deployment, image publishing, and inference engine updates.
  • Accelerate research on scaling test-time compute, RL rollout, and model-hardware co-design for next-generation systems.

Basic Qualifications

  • Deep low-level systems programming (C/C++ or Rust)
  • Experience with large-scale, high-concurrent production serving
  • Experience with GPU inference engines (vLLM, SGLang, Triton, TensorRT-LLM, etc.)
  • Strong background in system optimizations: batching, caching, load balancing, parallelism
  • Low-level inference optimizations: GPU kernels, code generation
  • Algorithmic inference optimizations: quantization, speculative decoding, distillation, low-precision numerics
  • Experience with testing, benchmarking, and reliability of inference services
  • Experience designing and implementing CI/CD infrastructure for inference

Compensation and Benefits

  • $180,000 - $440,000 USD
  • Equity compensation
  • Medical, vision, and dental coverage
  • 401(k) retirement plan
  • Short and long-term disability insurance
  • Life insurance
  • Additional discounts and perks

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

Similar jobs you might like

Technology

xAI

Software Engineer - Training/Inference (C++)

Senior

On-site

Palo Alto, CA

🏢 Summary: High-impact inference engineering role focused on building and optimizing large-scale distributed model serving systems for Grok. The position involves low-level GPU and inference optimization, scalable infrastructure development, and ensuring high-performance, reliable AI serving at massive scale. Candidates will work across the full inference stack, from orchestration to GPU kernels and CI/CD systems. 🗂️ Requirements: Deep systems programming in C/C++ or Rust, Experience with large-scale high-concurrency production serving, Experience with GPU inference engines, Strong knowledge of batching, caching, load balancing, and parallelism, Experience with GPU kernel optimization and code generation, Knowledge of quantization, speculative decoding, distillation, and low-precision numerics, Experience with testing, benchmarking, and reliability engineering for inference services, Experience designing and implementing CI/CD infrastructure for inference 📃 Skills: C, C++, Rust, vLLM, SGLang, Triton, TensorRT-LLM, GPU, CI/CD, Quantization, Distillation, Parallelism, Caching, Benchmarking, Load-balancing, Autoscaling 🏢 Description: SpaceXAI'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: We are building the high-performance inference platform that serves Grok to millions of users every day with lightning speed and perfect reliability. As a Member of Technical Staff - Inference, you will design and optimize large-scale model serving systems end-to-end. You will own everything from distributed infrastructure (global KV cache, continuous batching, load balancing, auto-scaling) to deep low-level optimizations (GPU kernels, quantization, speculative decoding, tail latency). This is a high-impact role where your work directly determines how fast and reliably users interact with Grok at massive scale. Responsibilities: - Architect and implement scalable distributed infrastructure for model serving (load balancing, auto-scaling, batch scheduling, global KV cache). - Optimize latency and throughput of model inference under real production workloads. - Build reliable, high-concurrency serving systems that serve billions of users with 100% uptime, 0% error rate, and excellent tail latency. - Benchmark, fine-tune, and accelerate inference engines (including low-level GPU kernel work and code generation). - Develop custom tools to trace, replay, and fix issues across the full stack — from orchestration down to GPU kernels. - Create robust CI/CD infrastructure for seamless endpoint deployment, image publishing, and inference engine updates. - Accelerate research on scaling test-time compute, RL rollout, and model-hardware co-design for next-generation systems. Basic Qualifications: - Deep low-level systems programming (C/C++ or Rust) - Experience with large-scale, high-concurrent production serving. - Experience with GPU inference engines (vLLM, SGLang, Triton, TensorRT-LLM, etc.). - Strong background in system optimizations: batching, caching, load balancing, parallelism. - Low-level inference optimizations: GPU kernels, code generation. - Algorithmic inference optimizations: quantization, speculative decoding, distillation, low-precision numerics. - Experience with testing, benchmarking, and reliability of inference services. - Experience designing and implementing CI/CD infrastructure for inference. Compensation and Benefits: - $180,000 - $440,000 USD - Base salary is just one part of the total rewards package, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks. SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.

Technology

SpaceXAI

Software Engineer - Kernels/CUDA (C++)

Senior

On-site

Palo Alto, CA

129,996 - 234,996 USD/yr

🏢 Summary: Role focused on building and optimizing large-scale GPU supercomputing infrastructure for AI training and inference. The position involves low-level GPU kernel optimization, distributed compute systems, and full-stack performance tuning across hardware and platform layers. Candidates will work closely with AI research teams to improve scalability, reliability, and training speed. 🗂️ Requirements: Deep low-level systems programming experience, Experience with large-scale GPU clusters, Hands-on GPU kernel optimization, Experience with distributed compute infrastructure, Ability to optimize memory-bound and compute-bound workloads, Strong communication skills 📃 Skills: CUDA, C, C++, PTX, SASS, CUTLASS, Nsight, Linux, GPU, Tensor 🏢 Description: SpaceXAI'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: We are building one of the world's largest AI supercomputers from the ground up. As part of the Compute Infrastructure team, you will own both the raw GPU supercomputer and the platform layer that runs on top of it. You will work across the full stack — from low-level GPU kernel optimizations and Linux kernel internals to massive-scale orchestration and virtualization — to make training and inference as fast, reliable, and scalable as possible. This is a broad, high-impact role that combines hardcore supercompute and compute infrastructure work. Your contributions will directly accelerate Grok's training speed and overall AI progress. RESPONSIBILITIES: - Design, build, and optimize massive GPU clusters for extreme-scale training and inference workloads - Develop and tune low-level CUDA kernels (GeMM, Attention, etc.), using CUTLASS, Tensor Cores, and Nsight for maximum performance - Profile, debug, and eliminate bottlenecks across GPU memory hierarchy, networking fabric, filesystems, and multi-GPU operation - Collaborate closely with AI research teams to deliver production-grade performance and scalability PREFERRED SKILLS AND EXPERIENCE: - Deep low-level systems programming (C/C++/PTX/SASS) - Strong experience with large-scale GPU clusters or distributed compute infrastructure at production scale - Hands-on work with GPU kernel optimization (CUTLASS, custom kernels, Nsight profiling) - Track record of building or running high-performance infrastructure for AI workloads (training or inference platforms) - Ability to reason from first principles and optimize for both memory-bound and compute-bound scenarios COMPENSATION AND BENEFITS: - $180,000 - $440,000 USD - Equity - Medical, vision, and dental coverage - 401(k) retirement plan - Short and long-term disability insurance - Life insurance - Additional discounts and perks SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.

Technology

SpaceXAI

Software Engineer - Kernels/CUDA (C++)

Senior

On-site

Seattle, WA

123,996 - 170,004 USD/yr

🏢 Summary: Role focused on building and optimizing large-scale AI supercomputing infrastructure for training and inference workloads. The position involves low-level GPU kernel optimization, distributed systems engineering, and full-stack compute infrastructure development. Candidates will work on performance, scalability, and reliability of massive GPU clusters supporting AI model training. 🗂️ Requirements: Deep systems programming experience in C/C++/PTX/SASS, Experience with large-scale GPU clusters or distributed compute infrastructure, Hands-on GPU kernel optimization experience, Experience with CUTLASS and Nsight profiling, Track record building high-performance AI infrastructure, Ability to optimize memory-bound and compute-bound workloads, Strong communication skills, Ability to work hands-on in a high-impact engineering environment 📃 Skills: CUDA, C, C++, PTX, SASS, CUTLASS, Nsight, Linux, GPU, Tensor, GeMM, Attention 🏢 Description: SpaceXAI'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: We are building one of the world's largest AI supercomputers from the ground up. As part of the Compute Infrastructure team, you will own both the raw GPU supercomputer and the platform layer that runs on top of it. You will work across the full stack — from low-level GPU kernel optimizations and Linux kernel internals to massive-scale orchestration and virtualization — to make training and inference as fast, reliable, and scalable as possible. This is a broad, high-impact role that combines hardcore supercompute and compute infrastructure work. Your contributions will directly accelerate Grok's training speed and overall AI progress. RESPONSIBILITIES: - Design, build, and optimize massive GPU clusters for extreme-scale training and inference workloads - Develop and tune low-level CUDA kernels (GeMM, Attention, etc.), using CUTLASS, Tensor Cores, and Nsight for maximum performance - Profile, debug, and eliminate bottlenecks across GPU memory hierarchy, networking fabric, filesystems, and multi-GPU operation - Collaborate closely with AI research teams to deliver production-grade performance and scalability PREFERRED SKILLS AND EXPERIENCE: - Deep low-level systems programming (C/C++/PTX/SASS) - Strong experience with large-scale GPU clusters or distributed compute infrastructure at production scale - Hands-on work with GPU kernel optimization (CUTLASS, custom kernels, Nsight profiling) - Track record of building or running high-performance infrastructure for AI workloads (training or inference platforms) - Ability to reason from first principles and optimize for both memory-bound and compute-bound scenarios COMPENSATION AND BENEFITS: - $180,000 - $440,000 USD - Base salary is part of a total rewards package including equity, comprehensive medical, vision, and dental coverage, 401(k) retirement plan, short and long-term disability insurance, life insurance, and additional discounts and perks. SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.

Technology

SpaceXAI

Member of Technical Staff - Voice Product

Senior

On-site

Palo Alto, CA

180,000 - 440,004 USD/yr

🏢 Summary: Backend-focused engineering role building and shipping scalable, low-latency voice AI features for Grok across iOS, Android, and other product surfaces. The position involves collaborating with model, media, and product teams to deliver high-quality voice experiences at global scale using Python or Rust. 🗂️ Requirements: Proficiency in Python or Rust, Backend engineering experience, Experience building scalable low-latency systems, Hands-on experience shipping features on iOS and/or Android, Strong interest in voice/audio AI, Ability to collaborate across product groups, Strong communication skills 📃 Skills: Python, Rust, iOS, Android, AI, Audio, Backend 🏢 Description: SpaceXAI'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: The Grok Voice Product team builds seamless, world-class voice experiences for Grok across every platform and product surface. In this role, you will be responsible for designing, building, and shipping voice features across all surfaces. You will have the opportunity to collaborate across iOS, Android, and all product groups. RESPONSIBILITIES: - Own backend engineering for scalable, low-latency voice infrastructure and model integrations. - Collaborate directly with Grok Voice model, media, and product teams to deliver end-to-end experiences. - Drive performance, reliability, and quality of voice interactions at global scale. - Move fast across platforms and product groups as a generalist. BASIC QUALIFICATIONS: - Proficiency in Python or Rust, with a strong track record of writing clean, efficient, maintainable, and scalable code. - Deep interest in voice/audio AI. - Strong generalist engineer with backend focus. - Hands-on experience shipping features on iOS and/or Android. - The ability to work across all product groups and do whatever it takes for best user experience. PREFERRED SKILLS AND EXPERIENCE: - Background in AI-driven consumer products or media generation technologies. - Track record collaborating across engineering, research, and product teams to ship delightful features quickly. COMPENSATION AND BENEFITS: - $180,000 - $440,000 USD - Equity package - Comprehensive medical, vision, and dental coverage - 401(k) retirement plan - Short and long-term disability insurance - Life insurance - Additional discounts and perks SpaceXAI is an equal opportunity employer. For details on data processing, view our Recruitment Privacy Notice.

Technology

SpaceXAI

Member of Technical Staff - X Search

Senior

On-site

San Jose, CA

180,000 - 440,004 USD/yr

🏢 Summary: Engineering role focused on building and operating large-scale search infrastructure and ML-powered retrieval systems for X. The position involves developing search algorithms, training ranking models, and integrating Grok capabilities into search products using modern backend and machine learning technologies. 🗂️ Requirements: Experience with Python, Experience with Go, Experience with Rust, Experience with vector databases, Experience with search indices, Experience with large-scale search systems, Experience with production ML systems, Ability to build scalable search infrastructure, Strong ownership and problem-solving skills, Ability to work in fast-paced environments 📃 Skills: Python, Go, Rust, ML, Search, Retrieval, Ranking, Databases, Indexing, Grok 🏢 Description: SpaceXAI'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: The X Search team is responsible for building and operating the core search engine that powers discovery on X. We own the full end-to-end search stack — from real-time retrieval and indexing to ranking and the end-user search experience. Our goal is to create the most relevant, fast, and intelligent search experience on the internet by combining state-of-the-art information retrieval systems with Grok's reasoning capabilities. RESPONSIBILITIES: - Build search and retrieval algorithms that scale to millions of users - Build and operate large scale search engine and search infrastructure - Train state of the art ML ranking models - Integrate Grok in search systems and products BASIC QUALIFICATIONS: - Tech Stack: Python, Go, Rust - Have worked with vector databases, search indices, or similar data stores for search applications - Have worked on large-scale search systems - Have deep experience building or working with production ML systems - Take full ownership of problems and learn whatever you need to solve them - Can move quickly in a fast-paced environment with shifting priorities and tight deadlines COMPENSATION AND BENEFITS: - $180,000 - $440,000 USD - Base salary is just one part of the total rewards package, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks. SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.

Technology

SpaceXAI

Member of Technical Staff - X Search

Senior

On-site

San Francisco, CA

180,000 - 440,004 USD/yr

🏢 Summary: Engineering role focused on building and operating large-scale search systems powering discovery on X, including retrieval, indexing, ranking, and ML-driven search experiences. The position involves developing scalable search infrastructure, training ranking models, and integrating Grok reasoning capabilities into search products. Candidates should have strong experience with production ML systems and large-scale search technologies. 🗂️ Requirements: Experience with Python, Experience with Go, Experience with Rust, Experience with vector databases or search indices, Experience building large-scale search systems, Experience with production ML systems, Ability to take ownership of technical problems, Ability to work in fast-paced environments with shifting priorities 📃 Skills: Python, Go, Rust, ML, Search, Retrieval, Indexing, Ranking, Databases, Infrastructure, Grok 🏢 Description: SpaceXAI'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: The X Search team is responsible for building and operating the core search engine that powers discovery on X. We own the full end-to-end search stack — from real-time retrieval and indexing to ranking and the end-user search experience. Our goal is to create the most relevant, fast, and intelligent search experience on the internet by combining state-of-the-art information retrieval systems with Grok's reasoning capabilities. RESPONSIBILITIES: - Build search and retrieval algorithms that scale to millions of users - Build and operate large scale search engine and search infrastructure - Train state of the art ML ranking models - Integrate Grok in search systems and products BASIC QUALIFICATIONS: - Tech Stack: Python, Go, Rust - Have worked with vector databases, search indices, or similar data stores for search applications - Have worked on large-scale search systems - Have deep experience building or working with production ML systems - Take full ownership of problems and learn whatever you need to solve them - Can move quickly in a fast-paced environment with shifting priorities and tight deadlines COMPENSATION AND BENEFITS: - $180,000 - $440,000 USD - Base salary is just one part of our total rewards package, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks. SpaceXAI is an equal opportunity employer. For details on data processing, view our Recruitment Privacy Notice.

Technology

SpaceXAI

Member of Technical Staff - X Search

Senior

On-site

Palo Alto, CA

180,000 - 440,004 USD/yr

🏢 Summary: Engineering role focused on building and operating large-scale search infrastructure and intelligent retrieval systems powered by machine learning and Grok integration. The position involves developing scalable search algorithms, training ranking models, and improving end-to-end search experiences for millions of users in a fast-paced environment. 🗂️ Requirements: Experience with Python, Experience with Go, Experience with Rust, Experience with vector databases or search indices, Experience with large-scale search systems, Experience with production ML systems, Ability to build and operate search infrastructure, Strong ownership and problem-solving skills, Ability to work in fast-paced environments, Strong communication skills 📃 Skills: Python, Go, Rust, ML, Search, Retrieval, Indexing, Ranking, Grok, Databases 🏢 Description: SpaceXAI'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: The X Search team is responsible for building and operating the core search engine that powers discovery on X. We own the full end-to-end search stack — from real-time retrieval and indexing to ranking and the end-user search experience. Our goal is to create the most relevant, fast, and intelligent search experience on the internet by combining state-of-the-art information retrieval systems with Grok's reasoning capabilities. RESPONSIBILITIES: - Build search and retrieval algorithms that scale to millions of users - Build and operate large scale search engine and search infrastructure - Train state of the art ML ranking models - Integrate Grok in search systems and products BASIC QUALIFICATIONS: - Tech Stack: Python, Go, Rust - Have worked with vector databases, search indices, or similar data stores for search applications - Have worked on large-scale search systems - Have deep experience building or working with production ML systems - Take full ownership of problems and learn whatever you need to solve them - Can move quickly in a fast-paced environment with shifting priorities and tight deadlines COMPENSATION AND BENEFITS - $180,000 - $440,000 USD - Base salary is just one part of the total rewards package, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks. SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.

Technology

SpaceXAI

Member of Technical Staff - Imagine Safety

Senior

On-site

Palo Alto, CA

180,000 - 440,004 USD/yr

🏢 Summary: Product Safety Engineer role focused on building scalable safety systems for multimodal AI media generation, including content moderation, risk detection, and evaluation frameworks for Grok users worldwide. The position involves developing real-time safety infrastructure, collaborating across engineering and research teams, and ensuring responsible deployment of AI-generated images, video, and audio at global scale. Compensation includes competitive salary, equity, and comprehensive benefits. 🗂️ Requirements: Proficiency in Rust, Experience building production safety or content moderation systems at scale, Expertise in real-time detection systems or high-throughput AI data pipelines, Ability to deliver reliable, scalable systems for millions of users, Strong problem-solving skills for AI safety challenges, Commitment to responsible AI development, Experience with multimodal or generative AI safety, Knowledge of distributed systems and real-time inference infrastructure 📃 Skills: Rust, Kubernetes, AI, ML, Inference, Observability, Moderation, Pipelines, DistributedSystems, RedTeaming, Classifiers, Audio, Video, Images 🏢 Description: SpaceXAI'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: The Imagine team is redefining AI-driven media experiences for Grok users worldwide. As a Product Safety Engineer on the Imagine team, you will build the critical safety systems and infrastructure that ensure Grok's multimodal generation capabilities (images, video, audio, and beyond) are powerful, delightful, and responsibly deployed. You'll design and scale safeguards, detection systems, and evaluation frameworks that prevent harm while preserving creativity and user freedom. Your work will directly shape how millions of users experience generative media — balancing rapid innovation with rigorous safety, trustworthiness, and alignment with xAI's mission. If you're a driven engineer passionate about AI safety, content integrity, and building robust systems that protect users without stifling innovation, this is your opportunity to make a lasting impact at the frontier of multimodal AI. RESPONSIBILITIES: - Design and implement scalable safety systems for Grok's media generation platform, including real-time content moderation, risk detection, and safeguard enforcement for images, video, and audio. - Build infrastructure to measure, monitor, and mitigate safety risks such as harmful content, bias, deepfakes, intellectual property issues, and misuse at global scale. - Develop tools, pipelines, and evaluation frameworks that enable rapid iteration on safety policies in collaboration with researchers, product, and policy teams. - Architect robust feedback loops between user interactions, model outputs, and training data to continuously improve safety while maintaining high performance and low latency. - Own full-cycle development of safety features: from problem definition and prototyping to deployment, monitoring, incident response, and long-term refinement. - Partner closely with the Imagine engineering, research, and product teams to embed safety into core features and deliver delightful yet responsible user experiences. BASIC QUALIFICATIONS: - Proficiency in Rust, with a strong track record of writing clean, efficient, maintainable, and scalable code. - Experience designing and building production safety, trust & safety, or content moderation systems for consumer-facing products at scale. - Hands-on expertise developing real-time detection systems, data pipelines, or evaluation frameworks for high-throughput AI applications. - Proven ability to deliver robust, reliable solutions that reach millions of users while maintaining high standards of uptime and performance. - Strong problem-solving skills and a passion for turning complex safety challenges into practical, high-impact engineering solutions. - Deep enthusiasm for responsible AI development and a commitment to building systems that advance humanity's understanding while protecting users. PREFERRED SKILLS AND EXPERIENCE: - Experience with multimodal content safety (images, video, audio) or generative AI safety in production environments. - Familiarity with machine learning classifiers, safety evaluation, red-teaming, or adversarial testing for media generation models. - Background in distributed systems, real-time inference serving, Kubernetes, observability tools, or large-scale data infrastructure. - Track record collaborating across engineering, research, and policy teams to ship safety-critical features quickly and effectively. - Previous work on content moderation, anti-abuse, model alignment, or responsible AI at consumer AI products. COMPENSATION AND BENEFITS: - $180,000 - $440,000 USD - Base salary is just one part of the total rewards package, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks. SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.

Technology

xAI

Member of Technical Staff - X Search

Senior

On-site

San Francisco, CA

180,000 - 440,004 USD/yr

🏢 Summary: Role focused on building and operating large-scale search systems and ML-powered retrieval infrastructure for X. The position involves developing search algorithms, training ranking models, and integrating Grok capabilities into intelligent search experiences using modern backend technologies. 🗂️ Requirements: Experience with Python, Experience with Go, Experience with Rust, Experience with vector databases, Experience with search indices, Experience with large-scale search systems, Experience with production ML systems, Ability to build and operate search infrastructure, Strong ownership and problem-solving skills, Ability to work in fast-paced environments 📃 Skills: Python, Go, Rust, ML, Search, Retrieval, Indexing, Vector, Databases, Ranking, Infrastructure, Grok 🏢 Description: SpaceXAI'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: The X Search team is responsible for building and operating the core search engine that powers discovery on X. We own the full end-to-end search stack — from real-time retrieval and indexing to ranking and the end-user search experience. Our goal is to create the most relevant, fast, and intelligent search experience on the internet by combining state-of-the-art information retrieval systems with Grok's reasoning capabilities. RESPONSIBILITIES: - Build search and retrieval algorithms that scale to millions of users - Build and operate large scale search engine and search infrastructure - Train state of the art ML ranking models - Integrate Grok in search systems and products BASIC QUALIFICATIONS: - Tech Stack: Python, Go, Rust - Have worked with vector databases, search indices, or similar data stores for search applications - Have worked on large-scale search systems - Have deep experience building or working with production ML systems - Take full ownership of problems and learn whatever you need to solve them - Can move quickly in a fast-paced environment with shifting priorities and tight deadlines COMPENSATION AND BENEFITS - $180,000 - $440,000 USD - Base salary is just one part of the total rewards package, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks. SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.

Technology

xAI

Member of Technical Staff - X Search

Senior

On-site

San Jose, CA

180,000 - 440,004 USD/yr

🏢 Summary: Role focused on building and operating large-scale search and retrieval systems that power discovery on X, including indexing, ranking, and ML-driven search experiences. The position involves developing scalable search infrastructure, training ranking models, and integrating Grok capabilities into production search systems in a fast-paced engineering environment. 🗂️ Requirements: Experience with Python, Experience with Go, Experience with Rust, Experience with vector databases, Experience with search indices, Experience with large-scale search systems, Experience with production ML systems, Ability to build scalable search and retrieval algorithms, Strong ownership and problem-solving skills, Ability to work in fast-paced environments, Strong communication skills 📃 Skills: Python, Go, Rust, ML, Search, Retrieval, Indexing, Ranking, Grok, Databases 🏢 Description: SpaceXAI'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: The X Search team is responsible for building and operating the core search engine that powers discovery on X. We own the full end-to-end search stack — from real-time retrieval and indexing to ranking and the end-user search experience. Our goal is to create the most relevant, fast, and intelligent search experience on the internet by combining state-of-the-art information retrieval systems with Grok's reasoning capabilities. RESPONSIBILITIES: - Build search and retrieval algorithms that scale to millions of users - Build and operate large scale search engine and search infrastructure - Train state of the art ML ranking models - Integrate Grok in search systems and products BASIC QUALIFICATIONS: - Tech Stack: Python, Go, Rust - Experience with vector databases, search indices, or similar data stores for search applications - Experience working on large-scale search systems - Deep experience building or working with production ML systems - Ability to take full ownership of problems and learn whatever is needed to solve them - Ability to move quickly in a fast-paced environment with shifting priorities and tight deadlines COMPENSATION AND BENEFITS: - $180,000 - $440,000 USD - Equity compensation - Comprehensive medical, vision, and dental coverage - 401(k) retirement plan - Short and long-term disability insurance - Life insurance - Additional discounts and perks SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.