# Member of Technical Staff, Recommendation Systems

**Company:** [xAI](https://hotfix.jobs/companies/xai)
**Location:** Palo Alto, CA
**Role:** ML Engineering
**Salary:** $180k – $440k/yr
**Skills:** Kafka, ClickHouse, Spark, Recommender Systems, Deep Learning, JAX, PyTorch, CUDA, Data Pipelines, Machine Learning
**Posted:** 2026-03-07

> Develop and scale recommendation algorithms, ranking, and search systems using AI stacks and data infrastructure for a platform with 600M users. Requires expertise in recommender systems, deep learning frameworks like JAX/PyTorch, and tools like Kafka/Spark.

## Job Description

## What You'll Do
- Designing and architecting recommendation algorithms across various product surfaces
- Leverage all of xAI’s infra and AI stacks to dramatically enhance the user experience
- Write data pipelines and training jobs that continuously learn from product data
- Iterate and improve the algorithm by gathering user feedback in real time through experimentation
- Ensuring scalability and efficiency of machine learning systems

## Who You Are
- Knowledge of data infrastructure like Kafka, Clickhouse, and Spark
- Experienced in implementing recommender systems and/or deep learning applications at industrial scale
- Skilled in one or more DL software frameworks such as JAX or PyTorch
- Exceptional candidates may be experienced in writing CUDA kernels

## Annual Salary Range
$180,000 - $440,000 USD

## Benefits
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.

## Similar roles

- [Member of Technical Staff](https://hotfix.jobs/jobs/dd7c8bc4-741a-45d5-91a8-e6b1c4ad67e0) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff, Applied Research](https://hotfix.jobs/jobs/d19460df-bf56-4f87-aa95-5e7bdf40798f) - LlamaIndex - San Francisco, CA - $180k – $250k/yr
- [Member of Technical Staff, Inference](https://hotfix.jobs/jobs/70280e8d-9e99-4be1-a488-b13edc2d4080) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Member of Technical Staff](https://hotfix.jobs/jobs/544a5b36-3a56-45fa-89fe-0873120ece22) - Black Forest Labs - San Francisco, CA - $180k – $290k/yr
- [Member of Technical Staff - X Search](https://hotfix.jobs/jobs/2480c779-d772-4053-b432-64b3e48d4cd3) - xAI - Palo Alto, CA - $180k – $440k/yr

**Apply:** https://hotfix.jobs/jobs/72f276af-c92c-4742-b5d8-9ecdf6a8c5c1
**Canonical:** https://hotfix.jobs/jobs/72f276af-c92c-4742-b5d8-9ecdf6a8c5c1