Member of Technical Staff - RL Infrastructure

xAI
Full-time
Palo Alto, CA; San Francisco, CA
$180,000 - $440,000
Posted on a month ago

Job Description

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. The role focuses on building and maintaining frameworks for agent, data, and model evaluation tasks, building environments for AI agents, and improving alerts and metrics for large-scale RL jobs.

Responsibilities

  • Creating and maintaining frameworks for agent, data, and model evaluation tasks
  • Building environments for AI agents
  • Building tools for automating workflows
  • Improving alerts, metrics and error handling on large scale RL jobs
  • Refactoring existing agent, data, eval, training frameworks
  • Designing operation procedures and coding standards
  • Writing unit tests and CI/CD frameworks

Requirements

  • Experience building and maintaining frameworks 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 and CI/CD
  • Strong communication skills

Benefits

  • No benefits