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_description.job_card.30_days_ago
serp_jobs.job_preview.job_type
  • serp_jobs.job_card.full_time
job_description.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

    serp_jobs.job_alerts.create_a_job

    Ai Researcher • Palo Alto, California, United States

    Job_description.internal_linking.related_jobs
    AI Engineer & Researcher - Pretraining and Scaling

    AI Engineer & Researcher - Pretraining and Scaling

    Xai • Palo Alto, California, United States
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_30 • serp_jobs.job_card.promoted
    Machine Learning Research Engineer, Generative AI

    Machine Learning Research Engineer, Generative AI

    Apple Inc. • Cupertino, CA, United States
    serp_jobs.job_card.full_time
    Machine Learning Research Engineer, Generative AI.Cupertino, California, United States Software and Services.Apple is where individual imaginations gather together, committing to the values that le...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_30 • serp_jobs.job_card.promoted
    Remote Machine Learning Engineer - AI Trainer ($80-$120 per hour)

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

    Mercor • Santa Cruz, California, US
    serp_jobs.filters.remote
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_1_day • serp_jobs.job_card.promoted
    Staff AI Researcher, Foundation Models

    Staff AI Researcher, Foundation Models

    Verily Life Sciences • Mountain View, CA, United States
    serp_jobs.job_card.full_time
    Verily is a subsidiary of Alphabet that is using a data-driven approach to change the way people manage their health and the way healthcare is delivered. Launched from Google X in 2015, our purpose ...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_variable_days • serp_jobs.job_card.promoted
    ML Research Engineer — HW Co-Design for Real-Time AI

    ML Research Engineer — HW Co-Design for Real-Time AI

    Etched • Cupertino, CA, United States
    serp_jobs.job_card.full_time
    A cutting-edge AI hardware company in Cupertino seeks a candidate to conduct research on custom AI chips, focusing on optimizing model performance and researching novel architectures.Responsibiliti...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_variable_days • serp_jobs.job_card.promoted
    Artificial Intelligence Engineer (San Jose)

    Artificial Intelligence Engineer (San Jose)

    NextGenPros Inc • San Jose, CA, United States
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_1_day • serp_jobs.job_card.promoted
    AI Engineer & Researcher - Reasoning Post-training

    AI Engineer & Researcher - Reasoning Post-training

    Xai • Palo Alto, California, United States
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_30 • serp_jobs.job_card.promoted
    Staff Machine Learning Engineer, AI Platform

    Staff Machine Learning Engineer, AI Platform

    General Motors • Sunnyvale, CA, United States
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_30 • serp_jobs.job_card.promoted
    Reinforcement Learning Research Engineer

    Reinforcement Learning Research Engineer

    Strativ Group • Fremont, CA, United States
    serp_jobs.job_card.full_time
    Reinforcement Learning Research Engineer.A scaling, SOTA Generative AI Startup operating with a world class team (Founders have multiple prior exits) with talent from Open AI, IBM, MIT and several ...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_variable_days • serp_jobs.job_card.promoted
    Artificial Intelligence Engineer

    Artificial Intelligence Engineer

    NextGenPros Inc • San Jose, CA, United States
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_1_day • serp_jobs.job_card.promoted
    ML Research Engineer for Home Robotics & Embodied AI

    ML Research Engineer for Home Robotics & Embodied AI

    Sunday Robotics • Mountain View, CA, United States
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_variable_days • serp_jobs.job_card.promoted
    Sr. Software Engineer- AI / ML, AWS Neuron Distributed Training

    Sr. Software Engineer- AI / ML, AWS Neuron Distributed Training

    Amazon • Cupertino, CA, United States
    serp_jobs.job_card.full_time
    Annapurna Labs designs silicon and software that accelerates innovation.Customers choose us to create cloud solutions that solve challenges that were unimaginable a short time ago-even yesterday.Ou...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_30 • serp_jobs.job_card.promoted
    AIML - ML Researcher in Foundation Models, Responsible AI

    AIML - ML Researcher in Foundation Models, Responsible AI

    Apple • Cupertino, CA, United States
    serp_jobs.job_card.full_time
    Join us as we build world-class groundbreaking products for our customers! Apple's Data and ML Innovation team focuses on innovative technologies, methodologies, and research to enable fantastic us...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_30 • serp_jobs.job_card.promoted
    Foundational ML Researcher / Engineer — Remote, Equity

    Foundational ML Researcher / Engineer — Remote, Equity

    Pathway Genomics Corporation • Palo Alto, CA, United States
    serp_jobs.filters.remote
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_variable_days • serp_jobs.job_card.promoted
    Reinforcement Learning Architect

    Reinforcement Learning Architect

    Mumba Technologies, Inc. • Los Altos, CA, United States
    serp_jobs.job_card.full_time
    We are seeking an experienced AI Engineer with deep expertise in Reinforcement Learning (RL) to join our team as a Senior Staff Architect. In this role, you will be responsible for shaping the visio...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_1_day • serp_jobs.job_card.promoted
    Foundation Models Engineer — AI Research & HPC Innovator

    Foundation Models Engineer — AI Research & HPC Innovator

    Institute of Foundation Models • Sunnyvale, CA, United States
    serp_jobs.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...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_variable_days • serp_jobs.job_card.promoted
    Remote Machine Learning Researchers(PhD) - AI Trainer ($120-$120 per hour)

    Remote Machine Learning Researchers(PhD) - AI Trainer ($120-$120 per hour)

    Mercor • Santa Cruz, California, US
    serp_jobs.filters.remote
    serp_jobs.job_card.full_time
    Role Overview • • Mercor is driving a leading AI research initiative focused on benchmarking and enhancing model performance across a range of machine learning tasks. We are seeking Machine Learning R...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_1_day • serp_jobs.job_card.promoted
    AI Researcher for Mobility PhD

    AI Researcher for Mobility PhD

    AVA Consulting • Mountain View, CA, United States
    serp_jobs.job_card.full_time
    Job Title : AI Researcher for Mobility PhD.Our client, a major employer in the area, is looking for an.Client is seeking a PhD graduate in Computer Science, Electrical Engineering, Mechanical Engine...serp_jobs.internal_linking.show_more
    serp_jobs.last_updated.last_updated_30 • serp_jobs.job_card.promoted