Software Engineer - Data

Job Description

Job Description

ABOUT xAI

xAI'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 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. All employees are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.

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. Our Data team is responsible for ensuring that the models are trained on the right data, in the right form, at the right quality, across every phase of the training lifecycle. This includes partnering closely with acquisition teams to identify where valuable data can be sourced, determining what data is needed to improve model performance, and building the production pipelines and systems that transform raw inputs into high-quality training data at scale. We work at the intersection of software, data, infrastructure, and machine learning to ensure our models train effectively and reliably.

As a Software Engineer on xAI's Data team, you will be responsible for developing applications that power data acquisition, preparation, training, quality evaluation, and delivery for model training. You will provide the ability to run training in a reliable, scalable and repeatable manner. You will also provide visibility on training status and data lineage. You will work closely with acquisition teams, ML engineers, and data engineers to build a reliable data pipeline to run training at scale. The ideal candidate combines strong software engineering fundamentals and excellent coding practices.

• 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

• 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

• 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

$150,000 - $250,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.

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