AI Engineer

1 month ago


Bodega Bay, United States xAI Full time
Job DescriptionJob DescriptionAbout the Position

As a Reasoning engineer, you will build frameworks to improve the reasoning capability, build distributed reinforcement learning systems, techniques for inference time compute (e.g. tree search and planning), and develop environments for agents.

You will get exposure and will be expected to solve and take ownership of components across the entire stack.

Tech Stack
  • Python
  • JAX
  • Rust
Location

The role is based in the Bay Area [San Francisco and Palo Alto]. Candidates are expected to be located in the Bay Area or open to relocation.

Focus
  • Build robust and scalable distributed RL systems.
  • Optimize frameworks to enable complex inference-time reasoning.
  • Develop environments and harnesses for agents.
Ideal Experiences
  • Experienced with large-scale reinforcement learning systems.
  • Designing and implementing distributed systems.
  • Keeping up with state-of-the-art RL and inference time compute algorithms.
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:

  1. Coding assessment in a language of your choice.
  2. Systems hands-on: Demonstrate practical skills in a live problem-solving session.
  3. Project deep-dive: Present your past exceptional work to a small audience.
  4. Meet and greet with the wider team.

Our goal is to finish the main process within one week. We don't rely on recruiters for assessments. Every application is reviewed by a member of our technical team. All interviews will be conducted via Google Meet.

Annual Salary Range

$180,000 - $440,000 USD

California Consumer Privacy Act (CCPA) Notice



  • Bodega Bay, United States RhythmX AI Full time

    Job DescriptionJob DescriptionAbout RhythmX AIRhythmX AI is a generative AI-native health company driving a paradigm shift in hyper-personalized care. RhythmX AI's precision care platform helps physicians pioneer a new era of whole-person care through generative and predictive AI-powered copilots. An SAIGroup company, RhythmX AI leverages SAIGroup...

  • Product Manager

    1 month ago


    Bodega Bay, United States RhythmX AI Full time

    Job DescriptionJob DescriptionProduct Manager / Technical Product Manager, RhythmX AI Location: Palo Alto, CAAbout RhythmX AI [https://rhythmx.ai/]RhythmX AI is a generative AI-native health company driving a paradigm shift in hyper-personalized care. RhythmX AI's precision care platform helps physicians pioneer a new era of whole-person care through...


  • Bodega, California, United States Lily AI Full time

    Transformative AI Engineer Role at Lily AIWe are seeking an exceptional Senior AI Technical Lead to spearhead the development of our machine learning platforms. This visionary role will empower you to shape the future of retail AI and drive business impact through scalable, high-performance solutions.

  • AI Finance Leader

    2 weeks ago


    Bodega, California, United States Perplexity AI Full time

    Drive Financial Growth at Perplexity AIWe are seeking an experienced Finance professional to join our team as a Controller. This role will be responsible for leading our financial operations, including accounting, financial reporting, and process improvements.Key Responsibilities:Financial Leadership: Develop and implement financial strategies aligned with...


  • Bodega Bay, United States Lily AI Full time

    Job DescriptionJob DescriptionAbout Lily AI: Lily AI is a female-founded retail AI company empowering retailers and brands by bridging the gap between merchant-speak and customer-speak. Leveraging computer vision, natural language processing, machine learning, and vertical-specific large language models (LLMs), the Lily platform enhances customer shopping...


  • Bodega Bay, United States Lily AI Full time

    Job DescriptionJob DescriptionAbout Lily AI: Lily AI is a female-founded retail AI company empowering retailers and brands by bridging the gap between merchant-speak and customer-speak. Leveraging computer vision, natural language processing, machine learning, and vertical-specific large language models (LLMs), the Lily platform enhances customer shopping...

  • AI Safety Engineer

    1 month ago


    Bodega Bay, United States xAI Full time

    Job DescriptionJob DescriptionAbout xAIxAI'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. Engineers are...


  • Bodega, California, United States MindsDB Full time

    AI Engineering ExcellenceMindsDB is a pioneer in the field of Artificial Intelligence, seeking an exceptional Lead AI Software Engineer to join our team. With a strong focus on innovation and collaboration, we empower companies to easily build AI capabilities that can Think, Understand and Orchestrate.About MindsDBWe are a fast-growing AI startup...


  • Bodega, California, United States Yurts Full time

    Unlocking Human-Centered AIWe are seeking a seasoned Engineering Manager to join our team at Yurts.ai, where we're redefining how teams harness AI for productivity and performance. This is an exceptional opportunity to lead the development of cutting-edge applied machine learning solutions that empower our customers.

  • AI / ML Developer

    3 days ago


    Bodega Bay, United States DuploCloud Full time

    Job DescriptionJob DescriptionAbout DuploCloud: DuploCloud is a software platform that allows engineering teams to achieve their infrastructure automation, security and compliance goals by offering DevOps-as-a-Service. We strive to make DevOps and Infrastructure-as-Code accessible for everyone; where application developers can focus on building applications...

  • AI Frontend Engineer

    2 weeks ago


    Bodega, California, United States MindsDB Full time

    About UsMindsDB is a pioneering AI startup headquartered in San Francisco, California. Our mission is to empower companies to easily build AI capabilities that can Think, Understand and Orchestrate, enabling teams to move from prototyping & experimentation to production in a fast & scalable way.Founded in 2017 by Adam Carrigan and Jorge Torres, MindsDB...


  • Bodega, California, United States xAI Full time

    About the RoleWe are seeking a highly skilled AI Safety Engineer to join our team at xAI. As an AI Safety Engineer, you will play a crucial role in ensuring the safety and reliability of our AI systems.

  • AI Safety Engineer

    2 weeks ago


    Bodega, California, United States xAI Full time

    Job OverviewxAI is a cutting-edge organization focused on developing AI systems that can accurately understand the universe and contribute to humanity's pursuit of knowledge. As an AI Safety Engineer, you will play a crucial role in ensuring the safety and reliability of our AI systems.About xAIOur team is highly motivated, with a flat organizational...


  • Bodega Bay, United States Redis Full time

    Job DescriptionJob DescriptionWho we areWe're Redis. We built the product that runs the fast apps our world runs on. (If you checked the weather, used your credit card, or looked at your flight status online today, you're welcome.) At Redis, you'll work with the fastest, simplest technology in the business—whether you're building it,...


  • Bodega, California, United States Lily AI Full time

    Unlock Your Potential as a Senior Machine Learning Architect at Lily AIAbout Us:Lily AI is a pioneering retail AI company founded by women, dedicated to empowering retailers and brands through innovative solutions. By bridging the gap between merchant-speak and customer-speak, our platform enhances shopping experiences, leveraging computer vision, natural...


  • Bodega, California, United States Yurts Full time

    Secure and Scalable Enterprise SolutionsAbout Us:At Yurts, we're harnessing the power of AI to unlock secure, knowledge platforms that serve people wherever they work. Born from a partnership with the Department of Defense, our platform is designed to meet the demands of mission-critical environments.We redefine how teams leverage AI for productivity and...


  • Bodega, California, United States DuploCloud Full time

    Job Summary:Audacious AI/ML Developer sought by DuploCloud Inc. to spearhead the development and optimization of machine learning models for our DevOps-as-a-Service platform. As a key member of our team, you will design, develop, and deploy cutting-edge AI solutions that drive business growth and customer satisfaction.About Us:DuploCloud is a pioneering...


  • Bodega, California, United States Lily AI Full time

    OverviewLily AI, a pioneering retail AI company, empowers retailers and brands by bridging the gap between merchant-speak and customer-speak. Our cutting-edge platform leverages computer vision, natural language processing, machine learning, and large language models to enhance customer shopping experiences.We are seeking an experienced Software Architect to...


  • Bodega Bay, United States MindsDB Full time

    Job DescriptionJob DescriptionABOUT USMindsDB is a fast-growing AI startup headquartered in San Francisco, California. As a leading innovator bringing AI and Data together, our passion is empowering companies to easily build AI capabilities that can Think, Understand and Orchestrate: enabling teams to move from prototyping & experimentation to production in...

  • Backend Engineer

    1 month ago


    Bodega Bay, United States MindsDB Full time

    Job DescriptionJob DescriptionABOUT USMindsDB is a fast-growing AI startup headquartered in San Francisco, California. As a leading innovator bringing AI and Data together, our passion is empowering companies to easily build AI capabilities that can Think, Understand and Orchestrate: enabling teams to move from prototyping & experimentation to production in...