GrepJob
xAI

ML Infrastructure Engineer

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

Base Salary

$180k - $440k/yr

Responsibilities

  • Design and build GPU compute infrastructure and training frameworks.
  • Develop data pipelines and integrate large-scale data systems.
  • Collaborate with ML teams to productionize models.
  • Ensure scalability, reliability, and efficiency of ML systems.
  • Solve complex problems independently across the full stack.
  • Mentor junior engineers and contribute to team growth.

Requirements

  • Bachelor's, Master's, or PhD in computer science, machine learning, or a related field.
  • 2+ years of experience in high traffic or large-scale production environments.
  • 2+ years of experience with ML platforms and training infrastructure.
  • Strong proficiency in Python and experience with C++ or Rust.

Benefits

  • Equity in the company.
  • Comprehensive medical, vision, and dental coverage.
  • Access to a 401(k) retirement plan.
  • Short & long-term disability insurance.
  • Life insurance and various discounts and perks.

Tech Stack

Categories

AI & MLBackendData EngineeringFull Stack