Talent.com
AI/ML Engineer (RL & Physical Systems)
AI/ML Engineer (RL & Physical Systems)Atlas • San Francisco, CA, United States
AI / ML Engineer (RL & Physical Systems)

AI / ML Engineer (RL & Physical Systems)

Atlas • San Francisco, CA, United States
[job_card.variable_days_ago]
[job_preview.job_type]
  • [job_card.full_time]
[job_card.job_description]

This range is provided by Atlas. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$150,000.00 / yr - $200,000.00 / yr

Direct message the job poster from Atlas

Connecting top talent with deep technology & manufacturing firms | Ex-Meta SWE

Atlas has partnered with Fluix AI to fill its Site Reliability Engineer role. To learn more about Fluix AI, you can review their website here.

FLUIX is building the AI Operating System for data centers. We deploy autonomous AI that optimizes, predicts, and controls AI factories.

Based in the San Francisco Bay Area, we develop intelligent control systems that enable data centers and power providers to operate faster, cleaner, and more efficiently.

Our mission is simpl e : help clients doubl e their compute capacity without wasting resources.

The Role

We’re hiring an AI / ML Engineer (or AI Scientist, depending on experience) with deep reinforcement learning and physics-based modeling expertise. You’ll design, test, and deploy models that interact with the physical world, from thermal systems to power distribution, where milliseconds and megawatts matter.

This is not a research-only position. Your work will touch real chillers, real cooling loops, and real megawatt-scale infrastructure.

Who You'll Work with

  • Abhi Sastri — Founder and CEO
  • Chase Overcash — CTO

What you’ll do

  • Design, develop, and deploy reinforcement learning–based control policies for real-world physical systems (cooling, power, airflow, thermodynamics, etc.).
  • Build and refine digital twin and simulation environments to accelerate training, testing, and Sim2Real deployment.
  • Conduct lab-based and field-based experiments to validate model performance under noisy, dynamic, and safety-critical conditions.
  • Analyze telemetry, time-series, and sensor data to evaluate model reliability, interpret failure cases, and propose improvements.
  • Support integration of LLM-based tools and workflows into the AI control pipeline where relevant (knowledge distillation, inference orchestration, etc.).
  • Lead or contribute to scientific documentation : whitepapers, internal reports, and peer-reviewed publications.
  • Push the frontier of physical-world AI, where physics, reinforcement learning, and industrial automation meet.
  • Collaborate with controls, software, and field engineering teams to integrate models into production-scale data centers and energy systems.
  • Your background

  • Bachelor’s degree required in Computer Science, Mechanical / Electrical Engineering, Applied Physics, Controls, or related field.
  • Master’s or Ph.D. strongly preferred for the AI Scientist tier.
  • 2+ years of hands‑on experience applying ML to real-world physical, robotic, industrial, or control systems.
  • Proficiency in Python and ML frameworks (PyTorch, TensorFlow); experience with RL libraries.
  • Strong grounding in at least one of : control theory, model‑predictive control (MPC), system identification, thermal / fluids, power systems, or industrial automation.
  • Experience working with telemetry / sensor data from PLCs, SCADA, IoT, or industrial control systems.
  • Familiarity with cloud or edge deployment (AWS / Azure, on‑prem GPUs, embedded compute).
  • Ability to move between research, experimentation, and deployment at startup speed.
  • Bonus Points

  • Experience deploying AI in data centers, utilities, industrial automation, HVAC, or energy systems.
  • Experience with digital twins, physics engines (Modelica, Simulink, custom simulators).
  • Publications, patents, or open-source work in RL, controls, or applied physical AI.
  • Experience with Sim2Real transfer, safety‑critical RL, or physics‑informed ML.
  • Experience with LLMs, agentic AI workflows, or hybrid RL + LLM systems.
  • Culture Fit

  • We are looking for obsessed builders who want their work to matter at physical scale.
  • You are energized by hard problems and high‑stakes environments.
  • You want to touch hardware, not just notebooks.
  • You believe AI belongs in the physical world, not just on cloud GPUs.
  • You thrive in “build it, ship it, iterate” environments rather than academic cycles.
  • Due to our mission‑critical work, you are eager to help teammates and co‑workers during holidays, weekends, and emergencies.
  • You are cordial and over‑communicate with teammates, co‑workers, and management.
  • Attractive compensation package, including equity options.
  • Company Benefits

  • Comprehensive health, dental, and vision insurance, along with other standard benefits.
  • A dynamic and collaborative San Francisco Bay Area work environment.
  • Opportunities for professional growth and development, with the chance to shape the future of technology in the industry.
  • Seniority level

  • Mid‑Senior level
  • Employment type

  • Full‑time
  • Industries

  • Energy Technology and Climate Data and Analytics
  • Referrals increase your chances of interviewing at Atlas by 2x.

    #J-18808-Ljbffr

    [job_alerts.create_a_job]

    System Engineer • San Francisco, CA, United States

    [internal_linking.similar_jobs]
    ML / AI Engineers

    ML / AI Engineers

    Adidev Technologies Inc • Daly City, CA, US
    [job_card.full_time]
    Salary : Competitive salary range : Based on experience and market value.This role is open to US Citizens, Green Card holders, GC-EAD only. Adidev is looking for an adept Machine Learning Engineer to ...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Senior ML Platform Engineer - Distributed AI Systems

    Senior ML Platform Engineer - Distributed AI Systems

    Apple Inc. • San Francisco, CA, United States
    [job_card.full_time]
    A leading technology company in San Francisco is seeking a Sr.Software Engineer to join the ML Platform Technologies team. You will design and build critical services for machine learning across App...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI / ML Software Engineer : Inference on Custom Accelerators

    AI / ML Software Engineer : Inference on Custom Accelerators

    Amazon • San Francisco, CA, United States
    [job_card.full_time]
    A leading e-commerce platform in San Francisco is seeking a Software Development Engineer to develop and optimize machine learning models for custom hardware accelerators.This role involves perform...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    ML / AI Research Engineer — Agentic AI Lab (Founding Team)

    ML / AI Research Engineer — Agentic AI Lab (Founding Team)

    Fabrion • San Francisco, CA, United States
    [job_card.full_time]
    ML / AI Research Engineer — Agentic AI Lab (Founding Team).Competitive salary + meaningful equity (founding tier).Backed by 8VC, we're building a world‑class team to tackle one of the industry’s most...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Senior AI Engineer – Foundational AI & LLM Systems

    Senior AI Engineer – Foundational AI & LLM Systems

    Capital One • San Francisco, CA, United States
    [job_card.full_time]
    A leading financial services company is seeking a Senior AI Engineer in San Francisco to develop responsible AI systems that transform customer interaction and enhance business operations.The ideal...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Hybrid ML Engineer — Physics AI & LLMs (Equity & Visa)

    Hybrid ML Engineer — Physics AI & LLMs (Equity & Visa)

    Apiphany Corporation • San Francisco, CA, United States
    [job_card.full_time]
    A pioneering AI company in San Francisco seeks an experienced ML Engineer to develop and implement advanced machine learning models. The ideal candidate will have expert-level programming skills in ...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI / ML Engineer

    AI / ML Engineer

    Krane • San Francisco, CA, United States
    [job_card.full_time]
    Get AI-powered advice on this job and more exclusive features.Krane is building intelligent tools that power the future of construction operations. You’ll lead the design and deployment of intellige...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    ML / AI Engineer

    ML / AI Engineer

    Rillet • San Francisco, CA, United States
    [job_card.full_time]
    Our customers are the financial brains of their companies.Our job is to help them run the numbers with impossible speed, accuracy, and insight. Today, we do that with powerful and elegant accounting...[show_more]
    [last_updated.last_updated_30] • [promoted]
    AI / ML Engineer

    AI / ML Engineer

    DeepAware AI (YC S25) • San Francisco, CA, United States
    [job_card.full_time]
    DeepAware AI (YC S25) is building secure, efficient, and autonomous infrastructure for the AI era.As an AI / ML Engineer, you’ll design, build, and deploy machine learning models that power our next-...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Senior ML Engineer : Real‑World AI & Production

    Senior ML Engineer : Real‑World AI & Production

    Ikuto • San Francisco, CA, United States
    [job_card.full_time]
    A venture-backed AI company in San Francisco is seeking a Machine Learning Engineer to develop and implement ML pipelines for real-world data. The role involves training language models and collabor...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    ML Engineer

    ML Engineer

    Phizenix • Menlo Park, CA, US
    [job_card.full_time] +1
    Client Opportunity | Through Phizenix.Phizenix, a certified minority and women-led recruiting firm, is hiring on behalf of an innovative generative AI startup that's developing diffusion-based ...[show_more]
    [last_updated.last_updated_30] • [promoted]
    AI Engineer, Applied ML

    AI Engineer, Applied ML

    Perplexity • San Francisco, California, United States
    [job_card.full_time]
    Perplexity is looking for an Applied ML Engineer to design, build, and iterate on cutting-edge AI models powering our core experience. As an expert in machine learning and artificial intelligence, y...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Robotics ML Research Engineer — Shape Physical AI

    Robotics ML Research Engineer — Shape Physical AI

    Scale • San Francisco, CA, United States
    [job_card.full_time]
    A leading AI technology company in San Francisco seeks an experienced Machine Learning Research Engineer to innovate in Robotics. The role involves developing ML pipelines and conducting applied res...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    ML / AI Software Engineer - Metrics Frameworks

    ML / AI Software Engineer - Metrics Frameworks

    General Motors of Canada • San Francisco, CA, United States
    [job_card.full_time]
    As a n AI / ML Engineer on the Metrics Frameworks team, part of the Simulation, Evaluation, and Data organization, you will be an individual contributor focused on developing and optimizing infrastru...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    ML Systems Research Engineer : Post-Training for AI

    ML Systems Research Engineer : Post-Training for AI

    Scale AI, Inc. • San Francisco, CA, United States
    [job_card.full_time]
    A leading AI technology firm in San Francisco seeks an ML Sys Research Engineer to optimize algorithms for their next-generation Agent RL training platform. The role involves building and profiling ...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    ML / AI Software Engineer - Metrics Frameworks

    ML / AI Software Engineer - Metrics Frameworks

    General Motors • San Francisco, CA, United States
    [job_card.full_time]
    As an AI / ML Engineer on the Metrics Frameworks team, part of the Simulation, Evaluation, and Data organization, you will be an individual contributor focused on developing and optimizing infrastruc...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI / ML Engineer

    AI / ML Engineer

    Rulebase • San Francisco, CA, United States
    [job_card.full_time]
    This range is provided by Rulebase.Your actual pay will be based on your skills and experience — talk with your recruiter to learn more. We are building an autonomous factory for financial service a...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI / ML engineer

    AI / ML engineer

    Orbofi • San Francisco, CA, United States
    [job_card.full_time]
    Orbofi is a pioneering AI-generated content engine, built specifically for Web3, games, and online communities.Our mission is to revolutionize the way content is created and consumed by harnessing ...[show_more]
    [last_updated.last_updated_30] • [promoted]