Member of Technical Staff - Recommendation Systems

xAI xAI · AI Frontier · Palo Alto, CA · Engineering

Seeking exceptional Applied engineers to work on a high-priority project with ~600 million monthly users, focusing on recommendation systems, ranking algorithms, and search technologies. Responsibilities include designing and architecting recommendation algorithms, leveraging AI stacks, writing data pipelines and training jobs, iterating through user feedback and experimentation, and ensuring scalability and efficiency of ML systems. Requires knowledge of data infrastructure (Kafka, Clickhouse, Spark), experience implementing recommender systems/deep learning at industrial scale, and proficiency in DL frameworks (JAX, PyTorch). CUDA kernel experience is a plus.

What you'd actually do

  1. Designing and architecting recommendation algorithms across various product surfaces
  2. Leverage all of xAI’s infra and AI stacks to dramatically enhance the user experience
  3. Write data pipelines and training jobs that continuously learn from product data.
  4. Iterate and improve the algorithm by gathering user feedback in real time through experimentation
  5. Ensuring scalability and efficiency of machine learning systems

Skills

Required

  • Kafka
  • Clickhouse
  • Spark
  • JAX
  • PyTorch

Nice to have

  • CUDA kernels

What the JD emphasized

  • industrial scale
  • recommendation systems
  • ranking algorithms
  • search technologies

Other signals

  • recommendation systems
  • ranking algorithms
  • search technologies
  • industrial scale
  • experimentation