Applied ML Engineer: Recommendation Systems & Real-Time AI

Xai; Palo Alto (CA); On-site; USD 180,000 - 440,000; Full time

Benefits offered by this job

Equity

Medical

Vision

Dental

401(k) retirement plan

Disability insurance

Life insurance

Discounts

Job summary

SpaceXAI is seeking exceptional Applied engineers to join a high-priority project used by hundreds of millions of monthly users. You will work at the intersection of advanced AI development and real-world impact, designing scalable recommendation and ranking systems.

You will collaborate with a small, hands-on team to build data pipelines, run experiments, and deploy efficient ML solutions that connect users with relevant content and experiences across products.

Qualifications

• Knowledge of data infrastructure (Kafka, ClickHouse, Spark).
• Experience implementing recommender systems or industrial-scale DL applications.
• Experience with DL frameworks such as JAX or PyTorch; CUDA kernel writing a plus.

Responsibilities

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

Skills

Kafka

ClickHouse

Spark

Recommender systems

Deep learning

JAX

PyTorch

CUDA kernels

Job description

SpaceXAI is seeking exceptional Applied engineers to join a high-priority project used by hundreds of millions of monthly users. You will work at the intersection of advanced AI development and real-world impact, designing scalable recommendation and ranking systems.

You will collaborate with a small, hands-on team to build data pipelines, run experiments, and deploy efficient ML solutions that connect users with relevant content and experiences across products.