Data Engineer

SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for individuals who appreciate challenging themselves and thrive on curiosity.

We operate with a flat organizational structure. All employees are expected to be hands-on and contribute directly to the mission. Leadership is given to those who show initiative and consistently deliver excellence.

About the Role

At xAI, we are building AI systems that push the frontier of human knowledge and scientific discovery. High-quality data is fundamental to every stage of that mission.

As a Data Engineer / AI Engineer on the Data team, you will develop systems, processes, and production code that power data acquisition, preparation, quality evaluation, and delivery for model training. You will collaborate closely with acquisition teams, ML engineers, and software engineers to build scalable data pipelines and improve the quality of training data.

Responsibilities

• Analyze the performance and impact of data used throughout the model training lifecycle
• Investigate anomalous model behavior and identify data issues affecting downstream performance
• Design and improve data cleaning, transformation, and quality-control systems
• Research and develop methods for improving data quality in AI model development
• Apply statistical techniques and empirical analysis to evaluate dataset quality
• Build and maintain production-grade data pipelines and tooling
• Develop metrics, evaluation frameworks, and monitoring systems for model behavior
• Fuse data from multiple sources into reliable datasets for model training
• Create shared datasets and internal tooling to improve model analysis and debugging

Basic Qualifications

• Bachelor's degree in computer science, data science, physics, mathematics, or STEM discipline
• 1+ years of data or software engineering experience
• Experience implementing or analyzing language models or neural networks

Preferred Skills and Experience

• Professional experience in analytics, data science, machine learning, or data engineering
• Experience building production data pipelines for large-scale machine learning workloads
• Strong Python experience and familiarity with ML libraries and tooling
• Experience with Parquet or similar columnar storage formats
• Familiarity with Kubernetes and distributed production environments
• Experience developing predictive models and ML pipelines including clustering, forecasting, and anomaly detection
• Experience working with terabyte- to petabyte-scale datasets
• Strong statistical and quantitative analysis skills
• Ability to manage evolving priorities in dynamic technical environments
• Ability to independently solve ambiguous technical problems

Compensation and Benefits

• $150,000 - $210,000 USD salary range
• Equity compensation
• Medical, dental, and vision coverage
• 401(k) retirement plan
• Short and long-term disability insurance
• Life insurance
• Additional discounts and perks

SpaceXAI is an equal opportunity employer. For details on data processing, view the Recruitment Privacy Notice.