AI in Robotics

Deep learning, vision and reinforcement learning for robots, built on Python, PyTorch, MuJoCo and Google Colab. Trainees add a layer of intelligence to the robot simulation they already know.

The second-stage track after the Aurora AI in Robotics foundation track. Aurora Robotics.

A small wheeled robot with a camera in a simulation lab
10

modules, one for each week

2

live sessions per week

2h

per session

10

portfolio projects, approximately

Program overview

A second-stage track for trainees who have completed Core 3 (Python, Linux and Git).

AI in Robotics teaches trainees to add a layer of intelligence to the robot simulation they already know, the Sensor and Actuator classes built in Core 3. It uses practical machine learning and computer vision and does not require heavyweight robotics or deep-learning tooling.

Every module builds on tools already used in Core 3, plus PyTorch for neural networks, MuJoCo for simulation and Gymnasium for reinforcement learning. Everything runs as a plain Python script on a normal laptop, so instructors and trainees work with a small, well-understood toolset throughout.

The program builds toward one running idea. Trainees collect or simulate sensor data, teach a simple model to make sense of it, and connect that decision back to the Actuator class. Each week adds one small piece to that pipeline, and the final project is a working AI-in-robotics program of the trainee's own design.

A neural network diagram next to a robot simulation

Learning outcomes

By the end of the program, trainees will be able to do the following.

Simulate robots

Simulate robots with MuJoCo.

Handle robot data

Collect, log and label their own robot data, and split it correctly (by run, not by row) to avoid leakage.

Build neural networks

Build, train and track PyTorch neural networks for robot control, and compare them fairly with classical models such as the random forest.

Fine-tune vision models

Fine-tune pretrained vision models (image classification and YOLO object detection) on custom images and use their outputs to drive robot behaviour.

Use language models

Use pretrained language and vision-language models (sentence embeddings, an LLM that returns JSON, and CLIP) to turn typed instructions into validated robot actions, and explain where VLAs fall short.

Train policies

Train imitation-learning and reinforcement-learning policies (Q-learning and PPO), and recognise and fix reward hacking.

Deliver a capstone

Design, evaluate against a baseline, document and present an end-to-end capstone project.

Who this is for

Prior deep learning, reinforcement learning or robotics-middleware experience is not assumed.

Foundation track graduates

Aurora Robotics trainees who completed the foundation AI in Robotics track, or who can show equivalent skills: working Python, pandas and scikit-learn, and a trained classifier on a robot dataset.

Practical learners

Trainees who want skills they can apply immediately in their own projects, with a portfolio of about ten small, working projects by the end.

Prerequisites

Completion of the Aurora AI in Robotics foundation track or equivalent: Python basics and a scikit-learn model already trained on the Wall-Following Robot dataset (random forest).

Format and setup

Everything runs in the browser and on a local Python install. A short setup guide goes out before Session 1, and the first 20 minutes of Session 1 are a live account-check clinic.

How each session runs

20 minConcept walkthrough
80 minHands-on lab
20 minRecap and Q&A

Program facts

  • Sessions2 per week, 2 hours each
  • DeliveryLive sessions in consecutive weeks
  • SupportAsync support on WhatsApp and Slack, including the Sunday after the final session for capstone submission

What to bring to Session 1

  • A Google account, for Google Colab and Google Drive.
  • A GitHub account, set up from the foundation track, with access to the course starter repository.
  • Free accounts on Weights and Biases (experiment tracking) and Roboflow (image labelling).
  • A modern browser, a stable internet connection, and a headset or earphones for breakout rooms. Any laptop that can run a video call can run this program, because the GPU comes from Colab.
  • The saved random forest model and the Wall-Following Robot dataset from the foundation track. The starter repository includes a backup copy.

Tools and platforms

Google Colab

Runs the notebooks in the browser and supplies a free GPU for training.

Python

The language for every lab, run as a plain script on a normal laptop.

PyTorch

Used for building and training neural networks.

MuJoCo

The physics simulator that hosts the robot.

Gymnasium

The interface for reinforcement learning environments.

Stable-Baselines3

The standard library used to train PPO agents.

Ultralytics YOLO

Pretrained and fine-tuned object detection.

Roboflow

Image labelling for the custom detector dataset.

Weights and Biases

Experiment tracking for training runs and sweeps.

GitHub

Home of the course starter repository and every weekly artifact.

Module-by-module curriculum

Each module has a goal, a weekly artifact and two sessions. Every lab ends with a committed notebook plus a short clip or screenshot.

Phase 1: From Datasets to a Running Robot

Weeks 1 and 2. Trainees put the foundation-track random forest in charge of a simulated robot, then train a neural network and compare the two fairly.

A robot simulation running in MuJoCo
1
Module 1, Week 1

From Dataset to Running Robot

Goal

See what happens when the random forest from the foundation track drives a robot instead of scoring a test set, and learn why that gap matters.

Weekly artifact

A closed-loop report: the random forest's success rate over 20 simulated runs, with a short diagnosis of the failures.

S1
What is AI and setup
Key concepts

AI vs robotics vs AI in robotics; everyday examples (robot vacuums, self-driving cars); what PyTorch and MuJoCo are

Hands-on lab

Install PyTorch and MuJoCo.

S2
MuJoCo basics and deploying the random forest
Key concepts

MuJoCo model and data; the simulation step loop; reading sensors; applying control inputs; resetting a run; loading a saved model

Hands-on lab

Load the foundation-track random forest and wire it in as a live controller. Run 20 trials and record the success rate.

2
Module 2, Week 2

Neural Networks for Control

Goal

Train the first deep-learning model of the program and judge it fairly against the random forest it replaces.

Weekly artifact

A PyTorch wall follower with tracked training runs and a fair benchmark against the random forest.

S3
Your first neural network
Key concepts

Neurons, layers and loss in plain language; gradient descent as nudging the weights; PyTorch tensors, Dataset and DataLoader, and the training loop; switching on Colab's free GPU; experiment tracking with W&B

Hands-on lab

Build a small multi-layer perceptron (MLP) on the wall-following data; train it; log loss and accuracy curves to W&B.

S4
Neural net vs random forest
Key concepts

Overfitting and early stopping; normalising inputs; changing one hyperparameter at a time; a fair comparison means offline metrics plus closed-loop success rate

Hands-on lab

Deploy the MLP in the simulator; run a small hyperparameter sweep in W&B; benchmark the MLP against the random forest on accuracy, success rate and speed; begin the mini-assignment.

Module 2 mini-assignment: Neural Wall Follower

Train a PyTorch neural network that drives the simulated robot along a wall. Compare it with your random forest on both test-set accuracy and closed-loop success rate (at least 20 runs each), and explain in a few sentences which one you would deploy and why.

Correct training and tracking40%
Fair closed-loop comparison35%
Clarity of explanation25%

Phase 2: Perception and Language

Weeks 3 to 5. The robot gets a camera and learns to classify images, detect objects and follow typed instructions.

A camera view with object detection boxes on a small robot
3
Module 3, Week 3

Vision with CNNs

Goal

Give the robot a camera and teach it to recognise what it sees using a pretrained network.

Weekly artifact

A short clip of the robot reacting to what its camera sees, plus a confusion matrix.

S5
Images as input for deep learning
Key concepts

Convolutions as pattern detectors (no maths); pretrained networks and transfer learning; data augmentation; getting camera frames out of the simulator

Hands-on lab

Write a script that drives the robot around and saves labelled camera frames (for example clear path, obstacle, goal marker); build an image dataset with at least 100 frames per class.

S6
Train and deploy a vision classifier
Key concepts

Fine-tuning a pretrained model (for example MobileNet or ResNet) on a small dataset; evaluating with a confusion matrix; generalising to lighting and texture changes; turning a class prediction into a drive command

Hands-on lab

Fine-tune the model on Colab's GPU; connect predictions to robot actions (stop, turn or go). Break it: change the arena lighting and colours and measure how far accuracy drops.

4
Module 4, Week 4

Object Detection

Goal

Move from what is in this image to where it is, and use the answer to steer.

Weekly artifact

A fine-tuned detector driving the robot toward a target.

S7
Detection with YOLO
Key concepts

Classification vs detection; bounding boxes and confidence thresholds; IoU and mAP explained in plain terms; running a pretrained YOLO model with Ultralytics

Hands-on lab

Run a pretrained YOLO model on simulator camera frames; draw the boxes; compare detections against ground truth and see where a pretrained model fails on simulated objects.

S8
Training your own detector
Key concepts

Labelling images in Roboflow; fine-tuning a small YOLO model; using box position and size for steering and rough distance

Hands-on lab

Label a small dataset of the target object (a pre-labelled backup set is provided); fine-tune YOLO in Colab; make the robot steer toward the target and stop at a set distance; begin the mini-assignment.

Module 4 mini-assignment: Detect and React

Fine-tune an object detector on a custom target in the simulator and write a controller that makes the robot approach it and stop at a set distance. Report detection precision and recall on a held-out set of images, and the robot's success rate over at least 20 trials.

Detector trained and evaluated40%
Working robot behaviour35%
Measured results25%
5
Module 5, Week 5

Language and Vision-Language-Action Models

Goal

Learn how robots can take instructions in plain language, and how vision-language-action (VLA) models connect what a robot sees and what it is told to what it does, using pretrained models rather than training them from scratch.

Weekly artifact

A language-commanded robot: a typed instruction becomes a validated robot action in simulation, plus a table showing how often it did the right thing across 20 test instructions.

S9
Language for robots (NLP)
Key concepts

Text as data: tokens and embeddings explained by analogy (similar meanings sit close together); intent classification; sentence embeddings for matching a command to a robot action; using a language model to turn free text into structured output (JSON); hallucination, ambiguity, and why every model output must be validated before it reaches an actuator

Hands-on lab

Build a command interpreter that maps instructions like "back up a little" or "turn toward the red box" to a small set of robot actions, first with sentence embeddings and nearest-neighbour matching, then with an LLM that returns JSON. Add a validator that rejects unknown or unsafe commands, and run the result on the simulated robot. Break it: try vague, contradictory and misspelled commands and record what gets through.

S10
Vision-language-action models
Key concepts

Vision-language models (image and text in, text out) vs VLAs (image and instruction in, action out); what models like CLIP, RT-2, OpenVLA and Octo do and how they were trained (survey level); zero-shot recognition; why large VLAs are often too heavy for free compute, and what to use instead; where VLAs fall short (speed, precision, safety)

Hands-on lab

Use CLIP to score the robot's camera frames against text prompts ("a red box ahead", "a clear path") and turn the scores into stop, turn or go decisions. Compare with a hand-coded colour rule on 20 frames. Watch a pre-recorded or hosted VLA demo and map its inputs and outputs onto your own pipeline.

Phase 3: Learning to Act

Weeks 6 to 8. Trainees compare a PID controller with a behaviour-cloning policy, then train agents from reward and audit what they learned.

Training curves for a reinforcement learning agent on a laptop screen
6
Module 6, Week 6

Imitation Learning and Control

Goal

Learn control from demonstrations, and see how it compares with the classic hand-tuned baseline.

Weekly artifact

A PID controller, a behaviour-cloning policy, and a side-by-side comparison of the two. A one-page capstone proposal is due at the end of the week.

S11
PID control: the baseline
Key concepts

P, I and D explained by feel; tuning by trial; why every learned controller should be compared with a simple baseline; behaviour cloning as supervised learning on observation and action pairs

Hands-on lab

Hand-tune a PID wall follower (or goal seeker) in the simulator; record its success rate, tracking error and smoothness.

S12
Behaviour cloning
Key concepts

Collecting demonstrations from the PID controller or teleoperation; training a policy that outputs continuous actions; compounding error, and fixing it with extra demonstrations from failure states (the idea behind DAgger)

Hands-on lab

Train a PyTorch behaviour-cloning policy and deploy it; compare with PID; collect extra demos where it fails and retrain. Submit the capstone proposal.

7
Module 7, Week 7

Reinforcement Learning Fundamentals

Goal

Understand learning from reward by building the simplest version yourself.

Weekly artifact

A tabular Q-learning agent with reward curves for two different reward designs.

S13
Learning from reward
Key concepts

Agent, environment, state, action and reward; episodes; exploration vs exploitation (epsilon-greedy); the Q-table as a cheat sheet

Hands-on lab

Implement tabular Q-learning from scratch on a Gymnasium grid world (for example FrozenLake); plot reward per episode.

S14
A robot task as an RL problem
Key concepts

Designing states and actions (discretising sensor readings); the Gymnasium reset and step interface; what makes a good reward

Hands-on lab

Wrap a small grid-navigation task with obstacles as a custom Gymnasium environment; train Q-learning; evaluate over 20 episodes; compare two reward functions.

8
Module 8, Week 8

Deep Reinforcement Learning

Goal

Train a neural-network policy with a standard library, and learn to audit what it actually learned.

Weekly artifact

A trained PPO agent, its tracked runs, and a short audit of one unintended behaviour.

S15
From Q-tables to PPO
Key concepts

Why tables break down with continuous sensor input; DQN and PPO at an intuitive level; the Stable-Baselines3 workflow; reward shaping and reward hacking

Hands-on lab

Train PPO with Stable-Baselines3 on a provided continuous-navigation environment to confirm the pipeline; log reward curves in W&B; save checkpoints to Drive or GitHub.

S16
Train and audit a robot agent
Key concepts

Observation and termination design; evaluating a policy across multiple seeds; Colab time-outs and checkpoints; sim-to-real gaps and domain randomisation (concepts only)

Hands-on lab

Train PPO for obstacle avoidance or goal reaching; deliberately hunt for a reward hack and fix the reward; compare with PID and behaviour cloning; begin the mini-assignment. Pre-trained checkpoints are provided as a fallback.

Module 8 mini-assignment: Train and Audit an RL Agent

Train a PPO agent for obstacle avoidance or goal reaching in the simulator. Evaluate it over at least 20 episodes, identify one unintended behaviour (a reward hack or failure mode), change the reward or observation to address it, and report before-and-after results.

Agent trained and tracked35%
Audit of the failure mode35%
Before-and-after evidence30%

Phase 4: Integration and Capstone

Weeks 9 and 10. Trainees design, build, evaluate and present an AI-in-robotics project of their own.

Two trainees presenting a robot demo to a class
9
Module 9, Week 9

Capstone Design and Beyond the Basics

Goal

Define your capstone and measure it honestly, and look at where robot AI is heading.

Weekly artifact

An approved capstone design (scope, success metric, baseline, test plan) and a first logged evaluation.

S17
Where robot AI is heading and capstone design
Key concepts

Survey (concepts only): vision-language and foundation models in robotics, learned vs classical components, safety and testing of learned systems, ethics; choosing a success metric and a baseline

Hands-on lab

Capstone design review in breakout rooms: finalise scope, success metric and baseline; write a test plan; set up the repository skeleton and W&B project.

S18
Capstone build session 1
Key concepts

Evaluating on unseen seeds and maps; common failure modes; keeping the pipeline running end to end

Hands-on lab

Build the core of the project with instructor check-ins (about 10 minutes per team); run and log a first evaluation against the baseline.

10
Module 10, Week 10

Capstone and Demos

Goal

Finish, document and present the capstone.

Weekly artifact

The finished capstone repository, README and demo recording.

S19
Capstone build session 2 and documentation
Key concepts

Fixing the biggest failure found in evaluation; writing a README that lets someone else run your project; recording a demo

Hands-on lab

Finish core functionality; run the final evaluation; write the README; record a 2 to 3 minute demo.

S20
Final demos and program wrap-up
Key concepts

Presenting results: what you built, how you measured it, and what failed; reflecting on the journey from random forest to autonomous system; where to go next (ROS 2, real hardware, sim-to-real, larger RL frameworks)

Hands-on lab

Each team presents (about 7 to 10 minutes, scaled to class size) followed by Q&A.

Capstone project

Teams of one or two are allowed. Trainees may propose their own project or pick one of the example directions.

End of Week 6

One-page capstone proposal due.

Week 9

Design review and first evaluation.

Final session

Live demos and Q&A.

Sunday after the final session

Final submission due.

Example directions

An autonomous delivery robot that navigates a maze and recognises its target.

A learned obstacle-avoidance agent trained with RL and compared against a rule-based or PID baseline.

A hybrid system: a classical planner combined with a learned perception module.

Deliverable

A single GitHub repository containing:

  • A Python project, built on the course starter repository, that runs in a simulated environment from start to finish.
  • At least one learned component from Modules 2 to 8: a neural network, a CNN or YOLO detector, a behaviour-cloning policy, or an RL policy.
  • A baseline (rule-based, PID or random forest) and an evaluation comparing it with the learned system, using a metric defined up front and measured over at least 20 runs.
  • Optionally, a state-estimation or planning component.
  • A README.md explaining what the project does, how to run it in Colab, and a summary of the results, including what failed.
  • A 2 to 3 minute demo recording, a link to the W&B project, and a clean Git history.
A trainee presenting a working robot demo

Present results honestly

The final demo covers what was built, how it was measured, and what failed. Teams present for about 7 to 10 minutes, scaled to class size, followed by Q&A.

Capstone grading rubric

Learned component works

25%

The model or policy is trained properly, tracked, and makes sensible decisions.

Closed-loop integration

20%

The full sense, decide and act pipeline runs end to end without manual intervention.

Evaluation rigour

25%

A clear metric, repeated trials, a baseline comparison, and honest reporting of failures.

Code quality and organisation

15%

Clear functions and structure, reproducible runs, sensible Git history.

README and demo clarity

15%

Easy for someone else to understand, run, and follow the demo.

How the program is assessed

Assessment blends three mini-assignments, one mid-program checkpoint and a capstone project, so trainees get feedback throughout. Every lab also ends with a small weekly artifact that counts toward participation.

Mini-assignments (Modules 2, 4 and 8)

30%

10% each, graded on correctness, measurement and following instructions.

Mid-program checkpoint

15%

A working Perceiving Robot that combines vision and state estimation.

Capstone project

40%

Graded against the full rubric in the capstone section.

Attendance, weekly artifacts and participation

15%

Includes engagement during the two demo sessions.