Member of Technical Staff - RL Infrastructure [data, evals, agent]

xAI•Palo Alto, CA

10d•Onsite

About The Position

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 engineers are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates. xAI is seeking experienced software engineers to create robust data pipelines, comprehensive evaluations for benchmarking LLMs, and automation frameworks to increase the productivity of researchers and engineers.

Requirements

Experience building and maintaining frameworks that are used by many engineers.
Experience in building high-performance sandboxes, virtual machines, and simulations.
Experience building full-stack apps for automating workflows and data visualization.
Experience in rapid iteration of research to production cycles.
Experience in test automation, CI/CD.

Responsibilities

Creating and maintaining frameworks for agent, data, and model evaluation tasks.
Building environments for AI agents.
Tools for automating common workflows.
Improving alerts, metrics and error handling on large scale RL jobs.
Refactoring existing agent, data, eval, training frameworks for better modularity.
Designing operation procedures and coding standards to streamline the transition from small scale experimentation to large scale RL training.
Writing unit tests, CI/CD frameworks to support rapid development cycles.