Applied AI Engineer

xAI is seeking an Applied AI Engineer to work on recommendation systems, ranking algorithms, and search technologies for a high-traffic product with 600M monthly users. The role involves designing and architecting recommendation algorithms, writing data pipelines and training jobs, and iterating based on user feedback. Candidates should have experience with recommender systems, deep learning frameworks like JAX or PyTorch, and data infrastructure like Kafka, Clickhouse, Spark.

• 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

• 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

• Experience writing CUDA kernels