Skip to content
xAIxAIPalo Alto, CA

Software Engineer, Data

Build scalable data platforms for AI model training pipelines, including acquisition, preparation, quality evaluation, and delivery. Collaborate with ML and data engineers; requires 1+ years software experience and strong coding in Python/Rust/etc.

240k – 280k/yr
On-site1+ YOEData Engineering

About the role

Responsibilities

  • Develop a highly reliable and scalable enterprise data platform to orchestrate data acquisition, preparation, training, quality evaluation, and delivery for model training
  • Create new features such as data lineage, visibility, and monitoring for end-to-end training that improve the quality of the data and model performance
  • Collaborate with peers on architecture, design, and code reviews
  • Build prototypes to prove out key design concepts and quantify technical constraints
  • Own all aspects of software engineering and product development
  • Deep dive into business problems, find efficient solutions and apply first principles thinking

Basic Qualifications

  • Bachelor's degree in computer science, data science, engineering, math, physics, or scientific discipline; OR 2+ years of professional experience building software in lieu of a degree
  • 1+ years of experience in application development, software engineering, data engineering, or data science

Preferred Skills and Experience

  • Programming experience in Python, Rust, Java, C#, Scala, Go or similar languages
  • Frontend experience in Angular, React, or similar JavaScript frameworks
  • Hands-on experience with Kubernetes and containerized deployments
  • Experience with Ray, AI training and orchestration
  • Experience with relational and non-relational databases, data lakes e.g. PostgreSQL, Iceberg, Clickhouse, or similar
  • Experience with data exploration tools like Grafana, Superset, or similar
  • Good understanding of version control, testing, continuous integration, build, deployment and monitoring
  • Good understanding of statistics, machine learning algorithms and frameworks

Compensation and Benefits

  • Annual Salary Range: $240,000 - $280,000 USD
  • Base salary is just one part of our total rewards package at xAI, 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

Skills

PythonRustJavaKubernetesRayPostgresIcebergClickHouseGrafanaSuperset

Similar roles

Data Engineering jobs
xAI

Data Engineer

xAIPalo Alto, CA

Build and maintain production data pipelines for AI model training, ensuring high-quality data through cleaning, transformation, and evaluation. Requires 1+ years experience with Python, ML, and neural networks; bachelor's in STEM.

240k – 280k/yr
On-site1+ YOEData Engineering
Reddit

Software Engineer - Data Movement Platform

RedditUnited States

Software engineer building and maintaining scalable data movement infrastructure using Spark, Flink, and Airflow to support ML and analytics workloads processing 100B+ daily events.

164k – 230k/yr
Remote2+ YOEData Engineering
Nuro

Software Engineer, Data Platform

NuroMountain View, CA

Builds scalable data infrastructure for autonomous driving ML systems, processing large-scale batch/streaming data from logs and simulations. Requires Python/C++ proficiency, data pipeline experience, and bachelor's in CS/EE; 1+ years experience.

160k – 241k/yr
On-site1+ YOEData Engineering
Coinbase

Analytics Engineer

CoinbaseUnited States

Build and maintain production data models and pipelines for Coinbase's Compliance Data Mart, ensuring regulatory-grade data quality and supporting audits and exams.

152k – 179k/yr
Remote2+ YOEData Engineering
Turquoise Health

Data Science Engineer, Analytics

Turquoise HealthSan Diego, CA

Build data pipelines, models, dashboards, and analyses to support product and business decision-making. Requires 2+ years of Python/SQL experience with data modeling, ETL tools, and AWS.

145k – 160k/yr
Remote2+ YOEData Engineering