Talent.com
AI Engineer & Researcher - RL Training Framework
AI Engineer & Researcher - RL Training FrameworkXai • Palo Alto, California, United States
AI Engineer & Researcher - RL Training Framework

AI Engineer & Researcher - RL Training Framework

Xai • Palo Alto, California, United States
30+ days ago
Job type
  • Full-time
Job description

About xAI

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 and researchers are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.

About the Team

Reasoning infrastructure team at xAI builds an end to end RL training framework to enable pretrain scale RL.

About the Role

In this role you might :

  • Design and implement state-of-the-art distributed RL systems
  • Profile, debug, and optimize system performance
  • Software and algorithm co-design with researchers

Ideal Experience

  • Experience building, debugging, and optimizing large scale distributed training systems
  • Experience building async RL training frameworks
  • Experience in inference systems
  • Proficiency in Python, Jax, or Rust
  • Exceptional candidates may have :

  • PhD with AI or other scientific domains
  • Strong knowledge of reinforcement learning techniques
  • Experience building infra for large-scale reinforcement learning and multi-agent reinforcement learning
  • Location

  • We hire engineers in Palo Alto. Our team usually works from the office 5 days a week but allow work-from-home days when required. Candidates are expected to be located near Palo Alto or open to relocation.
  • Interview Process

    After submitting your application, the team reviews your CV and statement of exceptional work. If your application passes this stage, you will be invited to a 15 minute interview (“phone interview”) during which a member of our team will ask some basic questions. If you clear the initial phone interview, you will enter the main process, which consists of four technical interviews :

  • Coding assessment in a language of your choice.
  • Researcher technical sessions (2) : These sessions will be testing your ability to formulate, design and solve concrete problems in real world with LLM. It can be research or engineering, depending on background / experience.
  • Meet the Team : Present your past exceptional work and your vision with xAI to a small audience.
  • Our goal is to finish the main process within one week. All interviews will be conducted via Google Meet.

    Annual Salary Range

    $180,000 - $440,000 USD

    xAI is an equal opportunity employer.

    California Consumer Privacy Act (CCPA) Notice

    Create a job alert for this search

    Ai Researcher • Palo Alto, California, United States

    Related jobs
    Senior AI Research Engineer, Model Inference (Remote)

    Senior AI Research Engineer, Model Inference (Remote)

    Tether Operations Limited • San Francisco, CA, United States
    Remote
    Full-time
    Join Tether and Shape the Future of Digital Finance.At Tether, we’re building solutions that empower businesses to integrate reserve-backed tokens across blockchains with transparency and trust in ...Show more
    Last updated: 30+ days ago • Promoted
    AI Research Lead - Pre-training (100% Remote)

    AI Research Lead - Pre-training (100% Remote)

    Tether Operations Limited • San Francisco, CA, United States
    Remote
    Full-time
    Join Tether and Shape the Future of Digital Finance.At Tether, we’re not just building products, we’re pioneering a global financial revolution. Our cutting-edge solutions empower businesses—from ex...Show more
    Last updated: 15 days ago • Promoted
    Research Engineer, Pre-training

    Research Engineer, Pre-training

    Anthropic • San Francisco, CA, United States
    Full-time
    Research Engineer, Pre-training.Get AI-powered advice on this job and more exclusive features.Anthropic’s mission is to create reliable, interpretable, and steerable AI systems.We want AI to be saf...Show more
    Last updated: 4 days ago • Promoted
    Foundational ML Researcher — Live AI, LLM Training, Remote

    Foundational ML Researcher — Live AI, LLM Training, Remote

    Pathway • Palo Alto, CA, United States
    Remote
    Full-time
    Pathway is seeking R&D Engineers to work on ambitious AI projects involving attention-based models.The role offers a competitive salary with an employee stock option plan and the opportunity to be ...Show more
    Last updated: 13 hours ago • Promoted • New!
    Reinforcement Learning Research Engineer

    Reinforcement Learning Research Engineer

    Strativ Group • San Francisco, CA, United States
    Full-time
    This range is provided by Strativ Group.Your actual pay will be based on your skills and experience — talk with your recruiter to learn more. A scaling, SOTA Generative AI Startup operating with a w...Show more
    Last updated: 4 days ago • Promoted
    ML Research Engineer - Training

    ML Research Engineer - Training

    Achira • San Francisco, CA, United States
    Full-time
    Join a world‑class team of scientists, ML researchers, and engineers working together to make the physical microcosm predictable and reshape the future of drug discovery. Move beyond the beaten path...Show more
    Last updated: 30+ days ago • Promoted
    Software Engineer, YAML - AI Training (Freelance, Remote)

    Software Engineer, YAML - AI Training (Freelance, Remote)

    Alignerr • San Francisco, CA, United States
    Remote
    Full-time
    Software Engineer, YAML - AI Training (Freelance, Remote).Software Engineer, YAML - AI Training (Freelance, Remote).Continue with Google Continue with Google. Software Engineer, YAML - AI Training (...Show more
    Last updated: 30+ days ago • Promoted
    Foundational ML Researcher - Live AI, LLM Training, Remote

    Foundational ML Researcher - Live AI, LLM Training, Remote

    Pathway • Palo Alto, CA, US
    Remote
    Full-time
    Pathway is seeking R&D Engineers to work on ambitious AI projects involving attention-based models.The role offers a competitive salary with an employee stock option plan and the opportunity to...Show more
    Last updated: 1 hour ago • Promoted • New!
    AI Engineer & Researcher, Inference - San Francisco, USA

    AI Engineer & Researcher, Inference - San Francisco, USA

    Speechify • San Francisco, CA, United States
    Full-time
    PLEASE APPLY THROUGH THIS LINK : https : / / job-boards.The mission of Speechify is to make sure that reading is never a barrier to learning. Over 50 million people use Speechify’s text-to-speech pr...Show more
    Last updated: 30+ days ago • Promoted
    Software Engineer, GDScript - AI Training (Freelance, Remote)

    Software Engineer, GDScript - AI Training (Freelance, Remote)

    Alignerr • San Francisco, CA, United States
    Remote
    Full-time
    Software Engineer, GDScript - AI Training (Freelance, Remote).Software Engineer, GDScript - AI Training (Freelance, Remote). Continue with Google Continue with Google.Software Engineer, GDScript - A...Show more
    Last updated: 30+ days ago • Promoted
    Senior AI Research Engineer, Model Inference (100% Remote)

    Senior AI Research Engineer, Model Inference (100% Remote)

    Tether Operations Limited • San Francisco, CA, US
    Remote
    Full-time
    Join Tether and Shape the Future of Digital Finance.At Tether, we’re not just building products, we’re pioneering a global financial revolution. Our cutting-edge solutions empower businesses—from ex...Show more
    Last updated: 30+ days ago
    Gen-AI UX Researcher

    Gen-AI UX Researcher

    US Tech Solutions • Sunnyvale, CA, United States
    Full-time
    In this role, you will develop and execute research programs that inform strategy across client Cloud’s Applied AI products. You will partner closely with UX, PM, Engineering, and Data Science to de...Show more
    Last updated: 1 day ago • Promoted
    AI Research Engineer

    AI Research Engineer

    Poly • San Francisco, CA, United States
    Full-time
    Poly is building a better file storage platform for everyone.We've raised $8M from YC, Bloomberg Beta, Felicis, Figma Ventures, AI Grant, and more, and we're looking to grow our founding team of su...Show more
    Last updated: 30+ days ago • Promoted
    Research Engineer - Distributed Training

    Research Engineer - Distributed Training

    Prime Intellect • San Francisco, CA, United States
    Full-time
    Building Open Superintelligence Infrastructure.Prime Intellect is building the open superintelligence stack - from frontier agentic models to the infra that enables anyone to create, train, and dep...Show more
    Last updated: 30+ days ago • Promoted
    Research Engineer, Pre-training : Scalable LLMs & Safe AI

    Research Engineer, Pre-training : Scalable LLMs & Safe AI

    Anthropic • San Francisco, CA, United States
    Full-time
    A leading AI research organization in San Francisco is seeking a Research Engineer for its Pre-training team.In this role, you will develop the next generation of large language models, combining c...Show more
    Last updated: 13 hours ago • Promoted • New!
    Researcher Engineer / Scientist, Training

    Researcher Engineer / Scientist, Training

    OpenAI • San Francisco, CA, United States
    Full-time
    OpenAI's Training team is responsible for producing the large language models that power our research, our products, and ultimately bring us closer to AGI. Achieving this goal requires combining dee...Show more
    Last updated: 30+ days ago • Promoted
    Machine Learning Research Lead, Security & Policy Research Lab

    Machine Learning Research Lead, Security & Policy Research Lab

    Scale AI, Inc. • San Francisco, CA, United States
    Full-time
    As the leading data and evaluation partner for frontier AI companies, Scale plays an integral role in understanding the capabilities and safeguarding AI models and systems.Building on this expertis...Show more
    Last updated: 30+ days ago • Promoted
    AI Research Lead

    AI Research Lead

    Perplexity AI Inc. • San Francisco, CA, United States
    Full-time
    Perplexity is seeking an exceptional AI Research Tech Lead to drive our research strategy and lead the development of our in-house Online LLMs, the Sonar models. In this leadership role, you will se...Show more
    Last updated: 7 days ago • Promoted