Machine Learning Engineer - Recommendation Systems
at xAI · 1001-5000 employees
Series E · $3B · Feb 2026
- Location
- Palo Alto, California
- Posted
- 5d ago
kafkasparkpytorch
at xAI · 1001-5000 employees
xAI is an advanced artificial-intelligence platform developer whose models (including Grok) are being integrated into large-scale infrastructure and, following its sale, SpaceX’s technology ecosystem.
<div class="content-intro"><p><span style="font-family: arial, helvetica, sans-serif;">SpaceXAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. </span><span style="font-family: arial, helvetica, sans-serif;">Our team is small, highly motivated, and focused on engineering excellence. This organization is for individuals who appreciate challenging themselves and thrive on curiosity. </span><span style="font-family: arial, helvetica, sans-serif;">We operate with a flat organizational structure. All employees are expected to be hands-on and to contribute directly to the company’s mission. Leadership is given to those who show initiative and consistently deliver excellence. Work ethic and strong prioritization skills are important. </span><span style="font-family: arial, helvetica, sans-serif;">All employees are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.</span></p></div><h3><strong>ABOUT THE ROLE:</strong></h3> <p>We’re seeking exceptional Applied 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, and many other systems. You’ll work at the intersection of advanced AI development and real-world impact, enhancing the ability to connect users with relevant content, accounts, and experiences.</p> <h3><strong>RESPONSIBILITIES:</strong></h3> <ul> <li>Designing and architecting recommendation algorithms across various product surfaces</li> <li>Leverage all of SpaceXAI's infra and AI stacks to dramatically enhance the user experience</li> <li>Write data pipelines and training jobs that continuously learn from product data.</li> <li>Iterate and improve the algorithm by gathering user feedback in real time through experimentation</li> <li>Ensuring scalability and efficiency of machine learning systems</li> </ul> <h3><strong>BASIC QUALIFICATIONS:</strong></h3> <ul> <li>Knowledge of data infrastructure like Kafka, Clickhouse, and Spark</li> <li>Experienced in implementing recommender systems and/or deep learning applications at industrial scale</li> <li>Skilled in one or more DL software frameworks such as JAX or PyTorch</li> <li>Exceptional candidates may be experienced in writing CUDA kernels</li> </ul> <h3><strong>COMPENSATION AND BENEFITS:</strong></h3> <p>$180,000 - $440,000 USD</p> <p>Base salary is just one part of our total rewards package at SpaceXAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.</p><div class="content-conclusion"><p><em>SpaceXAI is an equal opportunity employer. For details on data processing, view our </em><em><a href="https://x.ai/legal/recruitment-privacy-notice" target="_blank">Recruitment Privacy Notice</a>.</em></p></div>