Does MJX with Warp backend support parallel rendering?

[JunhongXu]

Intro

Hi!

I am using MJX for my research on training visual motor policies.

My setup

MuJoCo version 3.3.5, along with the latest MuJoCo warp.

My question

I am currently using the Warp backend for MJX. Does it support parallel GPU rendering?

Confirmations

• I searched the latest documentation thoroughly before posting.
• I searched previous Issues and Discussions, I am certain this has not been raised before.

[btaba]

It does not currently support parallel rendering beyond what is in ray, but we are working on it.

Also see https://github.com/google-deepmind/mujoco_playground and https://github.com/shacklettbp/madrona_mjx which are relevant