Machine Learning Engineer - Recommendation Systems

SpaceXSI is building high-priority recommendation and ranking/search technology for a product used by approximately 600 million monthly users. This onsite role in Palo Alto, CA is for applied engineers who want to translate product data into better experiences through algorithm design, continuous learning pipelines, and real-time experimentation. Compensation for this position ranges from $180,000 to $440,000 USD.

What you’ll work on

• Design and architect recommendation algorithms across multiple product surfaces.
• Use SpaceXSI’s infrastructure and AI stacks to improve user experience.
• Build data pipelines and training jobs that continuously learn from product data.
• Iterate on models by collecting user feedback through real-time experimentation.
• Ensure machine learning systems are scalable and efficient.

Skills and qualifications

• Knowledge of data infrastructure technologies such as Kafka, Clickhouse, and Spark.
• Experience implementing recommender systems and/or deep learning applications at industrial scale.
• Proficiency with one or more deep learning frameworks, including JAX or PyTorch.
• Exceptional candidates may have experience writing CUDA kernels.

Technologies you’ll use

• Kafka
• Clickhouse
• Spark
• JAX
• PyTorch
• CUDA

Benefits

• Equity
• Comprehensive medical, vision, and dental coverage
• Access to a 401(k) retirement plan
• Short and long-term disability insurance
• Life insurance
• Various other discounts and perks

Compensation: $180,000 - $440,000 USD per year