Machine Learning Engineer, Recommendation Systems

Job summary

Responsibilities

We’re seeking exceptional engineers to join a high-priority project that approximately 600 million monthly users use. This is an exciting opportunity for individuals with a engineer or scientist background to apply their skills to recommendation systems, ranking algorithms, search technologies…

• Designing and architecting recommendation algorithms across various product surfaces
• Leverage all of SpaceXSI'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

What they're looking for

• Experience in implementing recommender systems and/or deep learning applications at industrial scale
• Experience in writing CUDA kernels
• Skilled in one or more DL software frameworks such as JAX or PyTorch
• Knowledge of data infrastructure like Kafka, Clickhouse, and Spark

Tech stack

• CUDA
• JAX
• PyTorch
• Kafka
• Spark