Senior AI Software Engineer, Reinforcement Learning
$187k - $292kFull-time
Agility Robotics
Agility's commercially deployed humanoids operate alongside teams in warehouses, manufacturing facilities, and distribution centers—tackling physically demanding and repetitive tasks while enabling workers to focus on higher-value work. With industry-leading safety standards and years of proven deployment data, we're pioneering a new era of automation that enhances human potential. Agility is a pioneer. Our robot, Digit, is the first to be sold into workplaces across the globe. Our team is differentiated by its expertise in imagining, engineering, and delivering robots with advanced mobility, dexterity, intelligence, and efficiency -- robots specifically designed to work alongside people, in spaces built for people. Every day, we break through engineering challenges and invent new solutions and capabilities that will one day make robots commonplace and approachable. This work is our passion and our responsibility: our mission is to make businesses more productive and people's lives more fulfilling. Why Join Now Digit is one of the only humanoid robots deployed in real customer facilities today—at Schaeffler, GXO, Toyota Motor Manufacturing Canada, and Mercado Libre, with 65,000+ hours of real-world operation logged. In June 2026 we announced plans to become the first pure-play humanoid robotics company to trade publicly (ticker AGLT), in a ~$2.5B merger backed by NVIDIA, Amazon, SoftBank Vision Fund 2, and Foxconn. With $300M+ in orders for Digit and a factory built for 10,000 robots a year, we're scaling fast—and the policies you build will ship to robots working alongside people in production, not sit in simulation. About the Role The AI Controls team builds high-rate learned controllers that let Digit move robustly, efficiently, and safely in dynamic environments. As an AI Controls Engineer, you'll develop and deploy reinforcement learning policies across humanoid locomotion, whole-body control, and manipulation—integrating perception to enable collision-free, perceptive motion in the real world. About The Work
- Design, train, and deploy robust RL policies for locomotion, manipulation, whole body control, and dynamic interactions with the environment.
- Integrate perception into RL policies to achieve obstacle-aware, collision-free motion, and perceptive manipulation.
- Develop and maintain core RL infrastructure, including scalable training pipelines and evaluation frameworks.
- Design and implement new simulation environments and tasks to support training and evaluation of control policies.
- Collaborate with on-robot software and deployment teams to ship production-quality policies to Digit.
- 4+ years of experience developing and deploying RL policies for robotics applications.
- Strong Python skills and hands-on experience with a deep learning framework such as PyTorch.
- Experience designing reward functions, tuning hyperparameters, and implementing exploration strategies to solve complex control tasks.
- Experience with perception-in-the-loop control, integrating real-time sensory inputs for reactive or adaptive behaviors.
- Proven experience deploying reinforcement learning policies on real-world bipedal or quadrupedal robots.
- Familiarity with robot simulation environments (e.g. Mujoco-Warp, Isaac) and sim-to-real transfer.
- A collaborative approach and the ability to deliver safe, high-quality software in a fast-paced environment.
- Advanced degree (MS or PhD) in Robotics, Computer Science, or a related field.
- Experience with contact-rich manipulation, including force-torque or tactile sensing.
- Familiarity with policy distillation (e.g. teacher–student) for transferring state-based policies to perception-driven ones.
- Publications in top ML or robotics conferences (e.g. NeurIPS, ICML, CoRL, RSS, ICRA).
- 401(k) Plan: Includes a 6% company match.
- Equity: Company stock options.
- Insurance Coverage: 100% company-paid medical, dental, vision, and short/long-term disability insurance for employees.
- Benefit Start Date: Eligible for benefits on your first day of employment.
- Well-Being Support: Employee Assistance Program (EAP).
- Time Off:
- Exempt Employees: Flexible, unlimited PTO and 12 company holidays, including a winter shutdown.
- Non-Exempt Employees: 10 vacation days, paid sick leave, and 12 company holidays, including a winter shutdown, annually.
- On-Site Perks: Catered lunches four times a week and a variety of healthy snacks and refreshments at our Salem and Pittsburgh locations.
- Parental Leave: Generous paid parental leave programs.
- Work Environment: A culture that supports flexible work arrangements.
- Growth Opportunities: Professional development and tuition reimbursement programs.
- Relocation Assistance: Provided for eligible roles.
- Annual Discretionary Bonus: Provided for eligible roles.
Vacancy posted 18 hours ago
Similar jobs that could be interesting for youBased on the Senior AI Software Engineer, Reinforcement Learning in California vacancy
$187k - $292k
...expertise in imagining, engineering, and delivering robots with... ...About the Role The AI Controls team builds high-rate learned controllers that let... ...you’ll develop and deploy reinforcement learning policies across... ...Collaborate with on-robot software and deployment teams to...SeniorSoftwareFull timeTemporary workRelocation packageFlexible hours$197.3k - $225.1k
AI Engineer 4 (AI Foundations: LLM Customization, Finetuning, Reinforcement Learning)At Capital One, we are creating responsible and reliable... ...test, deploy, and support AI software components including... ...are met* Mentor Principal and Senior Associates on scalable design...SoftwareFull timePart timeLocal area$229.9k - $262.4k
AI Engineer 5 (AI Foundations: LLM Customization, Finetuning, Reinforcement Learning)At Capital One, we are creating responsible and reliable AI systems, changing banking for... ..., develop, test, deploy, and support AI software components including foundation model training...SoftwareFull timePart timeLocal area$99k - $225k
Reinforcement Learning AI EngineerThe Opportunity:Are you an innovative and experienced artificial intelligence... ...science, and machine learning (ML) engineering to train, test, deploy, and maintain... ...production systems, following strong software engineering best practices.Contribute...SoftwareFull timeContract workPart timeWork at officeLocal areaRemote work- ...Figure is an AI robotics company developing autonomous general-purpose humanoid... .... We are looking for a Helix AI Engineer, Reinforcement Learning to develop learning systems that enable... ...improve learning systems Solid software engineering skills and ability to build...SoftwareFull timeWork at office
- An innovative AI startup in San Francisco is seeking a member of the AI technical staff to build superhuman... ...include scaling infrastructure for reinforcement learning and collaborating with product engineers on product design. Ideal candidates will have experience...SeniorVisa sponsorshipRelocation package
$400k
...Senior Reinforcement Learning Environment Engineer Location Menlo Park, CA (San Francisco Bay Area) Working arrangement... ...will build environments in which AI agents can learn, use tools, and... ...complex tasks. The work combines software engineering with reinforcement...SeniorSoftwareFull time- ...As a Vision Machine Learning / Software Engineer, you will play a key role in enhancing Mentium's deep learning library for vision AI models on our in-memory compute ultra-low power hardware. Collaborating closely with a small team of software and hardware experts, you...SeniorSoftwareFull timeFlexible hours
$230k - $260k
...centered robotics company developing AI-powered robots to support humanity... ...for the better. JOB SUMMARY The Senior Reinforcement Learning Engineer is a key, hands-on role focused on... ...practices in reinforcement learning and software development. Collaborate closely...SeniorSoftwareLocal area$130k - $220k
Santa Clara, CASoftware Engineering - Motion Planning /... ...HybridPlusAI is a Physical AI company pioneering AI-based virtual driver software for factory-built... ..., develop, and deploy learned behavior planning models... ...approaches such as reinforcement learning, behavior cloning...SeniorSoftwareFull time$151.8k - $265.35k
Senior AI / Machine Learning Engineer — Fraud DetectionThe OpportunityWe are in search of an experienced Senior AI/ML Engineer to develop and broaden... ...to production deployment and monitoring.Strong software/data engineering skills across ML, backend, and data infrastructure...SeniorSoftwareFull timeTemporary workWork at officeLocal areaWorldwide3 days per week$174.72k - $295.68k
...integrating advanced AI and autonomous driving... ...edge R&D in AI, machine learning, and smart... ...motion controllers using reinforcement learning, imitation learning... ...advancements in academic and engineering research for humanoid... ...in Python and strong software design skills.1-3+...SeniorSoftwareFull time$201.3k - $352.3k
...DescriptionIt all started when engineer Fred Luddy wrote code that... ...work. Today, ServiceNow is the AI control tower for business reinvention... ...best practices around software engineering architecture and processes... ...their assigned work location. Learn more here. To determine...SeniorSoftwareWork experience placementWork at officeImmediate startRemote workFlexible hours$168k - $247k
...investing heavily in deep reinforcement learning to move beyond classical... ...minutes.About the RoleAs a Senior/Staff Deep RL Engineer, you will design, train,... ...proficiency in using AI coding tools (e.g., Claude... ...Codex, Cursor) in the full software development lifecycle, including...SeniorSoftwareHourly payWork at officeLocal areaRemote workFlexible hours- ...Intelligence, we're revolutionizing software development with AI-powered formal verification. We've developed... ...the role Join our team as an AI Engineer and help us push the boundaries of... ...and interfaces to accelerate machine learning development Derive practical solutions...SoftwareContract work
- ...Primarily On-site We are seeking an AI Evaluation Engineer – Reinforcement Learning & Agents to build the... ...representations of complex enterprise software environments. Develop... ...credentials or publication history. Seniority The opportunity is open to exceptional...Software
$131.4k - $236k
...it, then feed what you learn running it back into the... ...re investing deeply in AI/ML, LLMs, and... ...skilled and motivated Senior Applied Machine Learning... ...expertise with strong software development and leadership... ...computing).Experience applying reinforcement learning or agentic...SeniorSoftware- ...full description below. At Petco , we are bridging the gap between cutting-edge AI research and real-world software applications. We are looking for a Machine Learning Engineer who acts as the bridge between Data Science and Software Engineering. In this role...SeniorSoftware
$184k - $287.5k
...unlimited potential of AI to define the next... ...for outstanding Senior High Performance AI Engineer to build groundbreaking... ...with NVIDIA's software stack to provide comprehensive... ...with deep learning frameworks and... ...multi-agent systems, reinforcement learning, or AI systems...SeniorSoftwareFull timeRemote work$242.25k - $285k
...the connected platform and AI-native ecosystem for private... ...Carta brings together the software, services, and legal infrastructure... ...WithYou’ll join Carta’s ML Engineering team, embedded in Carta Law... ..., preference optimization, reinforcement learning, and related methods, with...SeniorSoftwareFull timeContract work$250k - $320k
(Staff / Senior Staff) AI Infra Engineer, Inference Infrastructure About The Role... ...We're looking for a software engineer, reliability-minded... ...Work alongside machine learning researchers, engineers, and... ...reliably, fine-tuning and reinforcement learning for creating...SeniorSoftwarePermanent employmentFull time$152k - $241.5k
...looking for outstanding Senior High Performance AI Engineers to build the next generation... ...with internal NVIDIA software, model, and hardware teams... ...experience in AI systems, machine learning, compilers, high-... ...systems, coding agents, reinforcement learning, or modern AI inference...SeniorSoftwareFull timeRemote work- ...Very few people in AI can say this.... ...covers. About Our Engineering Teams The Driving... ...team develops the learning methods that turn... ...technical owner for reinforcement learning within the... ...Your day-to-day As a Senior / Staff Machine Learning... ..., with strong software engineering...SeniorSoftwareWork at officeRemote workVisa sponsorshipRelocation packageFlexible hoursShift work
$174k - $252k
...one or more of the following: reinforcement learning (e.g., sequential decision... ..., maintaining, or launching software products, and 1 year of experience... ...serving.Google's software engineers develop the next-generation... ...with new integrations in AI Mode on Maps.As a part of our...SeniorSoftware$110k - $220k
...Walmart Global Tech Applied AI Data Science team is looking... ...with data scientists, machine learning engineers, operations research... ...managers, pricing strategists, software engineers, and data engineers... ...classical ML, deep learning, reinforcement learning), causal inference,...SoftwareFull timeTemporary workPart timeRemote work$191k - $253k
...by Lattice OS, an AI-powered operating... ...Anduril’s premier software platform that enables... ..., and Test Engineering to solve some of the... ...are looking for a Senior AI Infrastructure... ...end-to-end machine learning platform that powers... ..., validation, and reinforcement learning alignment...SeniorSoftwareFull timeWork experience placementImmediate start- .... Responsibilities Senior Machine Learning Engineer — Agentic Search Atlassian... ...systems that help people and AI agents discover relevant... ...optimization, or reinforcement learning.Experience with distributed... ...of every team. Our software products help teams all over...SeniorSoftwareWork at officeLocal area
- Google in Mountain View, CA, is seeking a software engineer to contribute to AI-powered advertising and large-scale systems. The role involves... ..., ML infrastructure, and experience with deep learning or reinforcement learning, with opportunities to work across teams such...SeniorSoftware
$120.1k - $214.5k
...advanced data analytics and AI to cybersecurity, we use innovative... ...by integrating machine learning and LLMs. Working closely with data scientists and software engineers through data extraction, research... ..., optimization, or reinforcement learningOpen-source contributions...SeniorSoftwareMinimum wageFull timeWork experience placementWork at officeLocal areaRemote work$160k - $190k
...artificial intelligence (“AI”) solutions, and other... ...novel machine learning and deep learning algorithms... ...with other researcher engineers to prototype and... ...Qualifications:Experience with Reinforcement Learning, self-... ...a roughly 100-person software org inside a company of...SeniorSoftwareLocal areaWorldwideVisa sponsorship
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior AI Software Engineer, Reinforcement Learning. Be the first to apply!
Related searches
- ai developer California
- ai engineer remote California
- ai prompt engineer California
- senior ai engineer California
- ai engineer California
- software engineer amazon California
- information technology software engineer California
- federal - software developer California
- graduate software developer California
- rust software engineer California





