[Feature] Add TD-MPC2 optimization stepper

e189412

Select commit

Loading

Failed to load commit list.

Uh oh!

There was an error while loading. Please reload this page.

Meta CLA

Meta CLA Check

Auto Label PR (File-Based) on: pull_request_target 3

label

PR Label on: pull_request_target 2

add-label

Validate Test Partitioning on: pull_request 2

validate

Push Binary Nightly on: pull_request 40

build-wheel-unix (linux, ubuntu-22.04, 3.10, cp310-cp310, cpu, cpu)

build-wheel-unix (linux, ubuntu-22.04, 3.11, cp311-cp311, cpu, cpu)

build-wheel-unix (linux, ubuntu-22.04, 3.12, cp312-cp312, cpu, cpu)

build-wheel-unix (linux, ubuntu-22.04, 3.13, cp313-cp313, cpu, cpu)

build-wheel-unix (linux, ubuntu-22.04, 3.14, cp314-cp314, cpu, cpu)

build-wheel-unix (macos, macos-latest, 3.10, cp310-cp310, cpu, cpu)

build-wheel-unix (macos, macos-latest, 3.11, cp311-cp311, cpu, cpu)

build-wheel-unix (macos, macos-latest, 3.12, cp312-cp312, cpu, cpu)

build-wheel-unix (macos, macos-latest, 3.13, cp313-cp313, cpu, cpu)

build-wheel-unix (macos, macos-latest, 3.14, cp314-cp314, cpu, cpu)

build-wheel-windows (3.10, 3.10.3)

build-wheel-windows (3.11, 3.11)

build-wheel-windows (3.12, 3.12)

build-wheel-windows (3.13, 3.13)

build-wheel-windows (3.14, 3.14)

test-wheel-unix (linux, ubuntu-22.04, 3.10, cp310-cp310, cpu, cpu)

test-wheel-unix (linux, ubuntu-22.04, 3.11, cp311-cp311, cpu, cpu)

test-wheel-unix (linux, ubuntu-22.04, 3.12, cp312-cp312, cpu, cpu)

test-wheel-unix (linux, ubuntu-22.04, 3.13, cp313-cp313, cpu, cpu)

test-wheel-unix (linux, ubuntu-22.04, 3.14, cp314-cp314, cpu, cpu)

test-wheel-unix (macos, macos-latest, 3.10, cp310-cp310, cpu, cpu)

test-wheel-unix (macos, macos-latest, 3.11, cp311-cp311, cpu, cpu)

test-wheel-unix (macos, macos-latest, 3.12, cp312-cp312, cpu, cpu)

test-wheel-unix (macos, macos-latest, 3.13, cp313-cp313, cpu, cpu)

test-wheel-unix (macos, macos-latest, 3.14, cp314-cp314, cpu, cpu)

test-wheel-windows (3.10, 3.10.3)

test-wheel-windows (3.11, 3.11)

test-wheel-windows (3.12, 3.12)

test-wheel-windows (3.13, 3.13)

test-wheel-windows (3.14, 3.14)

upload-wheel-unix

upload-wheel-windows

Continuous Benchmark (PR) on: pull_request

Pin benchmark definitions

Prepare pinned benchmark environment

${{ matrix.device }} ${{ matrix.revision }} benchmark

Compare benchmark results

SOTA Tests on Linux on: pull_request 2

tests (3.10, 13.0, 1) / linux-job

tests (3.10, 13.0, 2) / linux-job

Unit-tests on Linux on: pull_request 34

test-setup-minimal (3.10) / linux-job

test-setup-minimal (3.14) / linux-job

tests-cpu (3.10, bulk) / linux-job

tests-cpu (3.10, collectors) / linux-job

tests-cpu (3.10, mp) / linux-job

tests-cpu (3.13, bulk) / linux-job

tests-cpu (3.13, collectors) / linux-job

tests-cpu (3.13, mp) / linux-job

tests-cpu (3.14, bulk) / linux-job

tests-cpu (3.14, collectors) / linux-job

tests-cpu (3.14, mp) / linux-job

tests-cpu-mid

tests-gpu (3.12, 13.0, 1) / linux-job

tests-gpu (3.12, 13.0, 2) / linux-job

tests-gpu (3.12, 13.0, 3) / linux-job

tests-gpu-distributed (3.12, 13.0) / linux-job

tests-olddeps

tests-optdeps

tests-optdeps-smoke (3.12, 13.0) / linux-job

tests-stable-gpu (3.12, 13.0, 1) / linux-job

tests-stable-gpu (3.12, 13.0, 2) / linux-job

tests-stable-gpu (3.12, 13.0, 3) / linux-job

tests-stable-gpu-distributed (3.12, 13.0) / linux-job

upload-codecov

Examples Tests on Linux on: pull_request 4

tests (3.12, 13.0, 1) / linux-job

tests (3.12, 13.0, 2) / linux-job

Habitat Tests on Linux on: pull_request

tests

Tutorials Tests on Linux on: pull_request 1

tests (3.12, 13.0) / linux-job

Generate documentation on: pull_request 1

build-docs (3.12, 12.8) / linux-job

upload

update-gallery-cache

Lint on: pull_request 3

python-source-and-configs / linux-job

c-source / linux-job

lint-done

LLM Tests on Linux on: pull_request

unittests-vllm

unittests-sglang

Build Linux Wheels on: pull_request 10

generate-matrix / generate

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / build-manywheel-py3_10-cpu

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / build-manywheel-py3_10-cuda13_0

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / build-manywheel-py3_10-cuda13_2

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / build-manywheel-py3_10-cuda13_4

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / upload / upload-manywheel-py3_10-cpu

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / upload / upload-manywheel-py3_10-cuda13_0

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / upload / upload-manywheel-py3_10-cuda13_2

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / upload / upload-manywheel-py3_10-cuda13_4

Build Windows Wheels on: pull_request 4

generate-matrix / generate

pytorch/rl / build-wheel-py3_10-cpu

pytorch/rl / upload / upload-wheel-py3_10-cpu

Build M1 Wheels on: pull_request 7

generate-matrix / generate

pytorch/rl / build-wheel-py3_10-cpu

pytorch/rl / upload / upload-wheel-py3_10-cpu

Build Aarch64 Linux Wheels on: pull_request 4

generate-matrix / generate

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / build-wheel-py3_10-cpu-aarch64

pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl, .github/scripts/pre-build-script.sh) / upload / upload-wheel-py3_10-cpu-aarch64

Libs Tests on Linux on: pull_request 4

unittests-brax

unittests-craftground

unittests-mujoco-playground

unittests-mjlab

unittests-botorch (3.10, 12.8) / linux-job

unittests-envpool

unittests-gendgrl

unittests-gym (3.10, 12.8) / linux-job

unittests-isaaclab

unittests-jumanji

unittests-libero

unittests-meltingpot

unittests-open_spiel

unittests-chess

unittests-unity_mlagents

unittests-minari

unittests-openx

unittests-pettingzoo

unittests-procgen

unittests-safety-gymnasium

unittests-robohive

unittests-roboset

unittests-sklearn (3.10, 12.8) / linux-job

unittests-smacv2

unittests-vd4rl

unittests-vmas

unittests-torch_geometric (3.10, 12.8) / linux-job

PyTorch Bot

Dr.CI

Post-merge benchmarks on: pull_request_target 1

Dispatch the full benchmark suite

Meta CLA / Meta CLA Check succeeded Oct 5, 2026 in 0s

Contributor License Agreement is valid!

View more details on Meta CLA

Loading

Re-running checks...