Software Engineer, Data
ABOUT XAIxAI’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.About The RoleAt 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.ResponsibilitiesDevelop a highly reliable and scalable enterprise data platform to orchestrate data acquisition, preparation, training, quality evaluation, and delivery for model trainingCreate new features such as data lineage, visibility, and monitoring for end-to-end training that improve the quality of the data and model performanceCollaborate with peers on architecture, design, and code reviewsBuild prototypes to prove out key design concepts and quantify technical constraintsOwn all aspects of software engineering and product developmentDeep dive into business problems, find efficient solutions and apply first principles thinkingBasic QualificationsBachelor'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 degree1+ years of experience in application development, software engineering, data engineering, or data sciencePreferred Skills And ExperienceProgramming experience in Python, Rust, Java, C#, Scala, Go or similar languagesFrontend experience in Angular, React, or similar JavaScript frameworksHands-on experience with Kubernetes and containerized deploymentsExperience with Ray, AI training and orchestrationExperience with relational and non-relational databases, data lakes e.g. PostgreSQL, Iceberg, Clickhouse, or similarExperience with data exploration tools like Grafana, Superset, or similarGood understanding of version control, testing, continuous integration, build, deployment and monitoringGood understanding of statistics, machine learning algorithms and frameworksCompensation And BenefitsAnnual Salary Range: $240,000 - $280,000 USDBase 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.If you are currently employed in the field, please tell us the name of your employer.If you are currently employed in the field, please tell us your role including your seniority level (e.g. Software Engineer II).If you have a public LinkedIn profile, please provide its URL.If you have a Google Scholar page, please provide its URL.In 100 words or less, tell us about a piece of work you are most proud of.