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
[job_card.30_days_ago]
[job_preview.job_type]
  • [job_card.full_time]
[job_card.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

    [job_alerts.create_a_job]

    Ai Researcher • Palo Alto, California, United States

    [internal_linking.similar_jobs]
    AI Engineer & Researcher - Pretraining and Scaling

    AI Engineer & Researcher - Pretraining and Scaling

    Xai • Palo Alto, California, United States
    [job_card.full_time]
    AI’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 excelle...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Principal AI Training Platform Architect

    Principal AI Training Platform Architect

    LinkedIn • Mountain View, California, United States
    [job_card.full_time]
    A leading professional network is seeking a Principal Staff Software Engineer to lead the development of AI training infrastructure. The role requires extensive experience in software design and pro...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI Research Engineer

    AI Research Engineer

    Code Metal • Palo Alto, CA, US
    [job_card.full_time]
    Our mission is to enable hardware deployment at the speed of software development.We are working towards automatic code transpilation and optimization for any hardware application.In this role, you...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Foundational ML Researcher — Live AI, LLM Training, Remote

    Foundational ML Researcher — Live AI, LLM Training, Remote

    Pathway • Palo Alto, CA, United States
    [filters.remote]
    [job_card.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.last_updated_variable_days] • [promoted]
    AI Engineer & Researcher - Reasoning Post-training

    AI Engineer & Researcher - Reasoning Post-training

    Xai • Palo Alto, California, United States
    [job_card.full_time]
    AI’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 excelle...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Remote Machine Learning Engineer - AI Trainer ($80-$120 per hour)

    Remote Machine Learning Engineer - AI Trainer ($80-$120 per hour)

    Mercor • Danville, California, US
    [filters.remote]
    [job_card.part_time]
    At Mercor, we’re building the talent engine that helps leading labs and research orgs move AI forward.Our latest initiative focuses on benchmarking and improving model performance and training spee...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Staff Machine Learning Engineer, AI Platform

    Staff Machine Learning Engineer, AI Platform

    General Motors • Sunnyvale, CA, United States
    [job_card.full_time]
    Remote : This role is based remotely but if you live within a 50-mile radius of Mountain View, you are expected to report to that location three times a week, at minimum. We are seeking an experience...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Senior AI Engineer – Foundational AI & LLM Systems

    Senior AI Engineer – Foundational AI & LLM Systems

    Capital One • San Jose, CA, United States
    [job_card.full_time]
    A leading banking technology firm in San Jose is seeking a Senior AI Engineer to drive the development of AI systems.The role involves partnering with diverse teams to create AI-enhanced products a...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Artificial Intelligence Engineer

    Artificial Intelligence Engineer

    NextGenPros Inc • San Jose, CA, United States
    [job_card.full_time]
    Location : San Jose, CA (Hybrid).Investigate and develop methods for.Design and experiment with LLM.SQL-based processing and advanced text preprocessing techniques. LangChain, LangGraph, LangSmith fo...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    ML Research Engineer for Home Robotics & Embodied AI

    ML Research Engineer for Home Robotics & Embodied AI

    Sunday Robotics • Mountain View, CA, United States
    [job_card.full_time]
    A tech innovation company in Mountain View, California, is seeking a Machine Learning Research Engineer.You'll design sophisticated robot learning algorithms to enhance dexterous manipulation in ho...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Foundational ML Researcher / Engineer — Remote, Equity

    Foundational ML Researcher / Engineer — Remote, Equity

    Pathway Genomics Corporation • Palo Alto, CA, United States
    [filters.remote]
    [job_card.full_time]
    A cutting-edge AI startup is seeking R&D Engineers for groundbreaking work in attention-based machine learning models.Candidates should have a strong research background and experience in model tra...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    LLM Research Engineer

    LLM Research Engineer

    Cypress HCM • Mountain View, CA, US
    [job_card.full_time]
    Design, train, and fine-tune large language models (e.GPT, LLaMA, PaLM) for various applications.Conduct research on cutting-edge techniques in natural language processing (NLP) and machine learnin...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Applied Researcher II (AI Foundations, LLM Core and Agentic AI)

    Applied Researcher II (AI Foundations, LLM Core and Agentic AI)

    Capital One • San Jose, California, USA
    [job_card.full_time] +1
    Applied Researcher II (AI Foundations LLM Core and Agentic AI).At Capital One we are creating trustworthy and reliable AI systems changing banking for good. For years Capital One has been leading th...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Senior AI Research Engineer

    Senior AI Research Engineer

    Deep Abacus • Fremont, CA, US
    [job_card.full_time]
    VC-backed Series B startup on a high-growth trajectory.Revolutionizing commerce through direct engagement with powerful conversation AI. Join a world-class team with opportunity for rapid career adv...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Foundation Models Engineer — AI Research & HPC Innovator

    Foundation Models Engineer — AI Research & HPC Innovator

    Institute of Foundation Models • Sunnyvale, CA, United States
    [job_card.full_time]
    A leading research laboratory in Sunnyvale is seeking a Machine Learning Engineer to develop innovative machine learning models that address real-world challenges. This role involves collaboration w...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Applied Research Engineer

    Applied Research Engineer

    Confidential • Hayward, CA, US
    [job_card.full_time]
    About the Company (Confidential).Our client is a cutting-edge AI research company specializing exclusively in.Series A from top-tier investors. Matrix Partners, Swift Ventures, Y Combinator, and AI ...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Senior Principal Engineer- End-to-End AI Training Framework

    Senior Principal Engineer- End-to-End AI Training Framework

    Bosch Group • Sunnyvale, CA, US
    [job_card.full_time]
    Invented for Life” drives us at Bosch and our vision of future mobility.Autonomous vehicles will change the way we move and at Bosch we are working on making this future a reality.We are now ...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Agentic AI Researcher - PhD

    Agentic AI Researcher - PhD

    KYYBA Inc • Mountain View, CA, US
    [job_card.full_time]
    Research and Implementation of the research for future mobility services.Developing innovative solutions that.AI / ML systems for mobility services. Conducting applied research in Agentic AI, includin...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]