Member Of Technical Staff - RL Infrastructure (AI)

TL;DR

Member Of Technical Staff - RL Infrastructure (AI): Creating robust data pipelines and automation frameworks to enhance productivity in AI research and engineering with an accent on building evaluation frameworks and environments for AI agents. Focus on designing operational procedures and coding standards for large-scale RL training.

Location: Must be based in the Bay Area or open to relocation

Company

xAI is focused on creating AI systems that understand the universe and aid humanity.

What you will do

• Develop and maintain frameworks for agent, data, and model evaluation tasks.
• Build environments for AI agents and automate common workflows.
• Improve alerts, metrics, and error handling on large-scale RL jobs.
• Refactor existing frameworks for better modularity.
• Write unit tests and CI/CD frameworks to support rapid development cycles.

Requirements

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