GrepJob
xAI

Member of Technical Staff - RL Inference

xAI
Apply
about 3 hours ago
Palo Alto, CA, USAMid Level / Senior
H1B Sponsor

Responsibilities

  • Design and optimize the inference stack for RL workloads.
  • Analyze and address performance bottlenecks in large scale RL systems.
  • Collaborate with the modeling team to implement novel RL techniques.

Requirements

  • Experience in building and optimizing large-scale distributed systems.
  • Experience in LLM inference.
  • Proficiency in programming languages such as Python, C++, and/or Rust.
  • Familiarity with frameworks such as PyTorch, Jax, and CUDA.
  • Willingness to solve complex problems at all levels of the stack.

Categories

AI & MLData Science