Real-time Web Viewer

[bsyh]

Hello Brax community,

This post presents a prototype for a real-time, web-based visualizer for Brax training, developed as part of Google Summer of Code 2025. This work aims to address the long-standing request for a live visualization feature, as discussed in issues #47 and #107.

Demo

Key Features

• Real-time Streaming: Renders Brax environment states during training.
• Web-based: Access the viewer from any web browser, perfect for remote headless training setups.
• JAX-compatible: Designed for minimal impact on training performance (currently optimized for PPO).
• Lightweight: Built with standard dependencies.
• Interactive Controls: Toggle rendering on the fly from the web interface to save compute resources.
• Distributed Architecture: The viewer server runs as an independent process from the training script.

How to Test

We invite the community to test this prototype. You can get started here:
https://github.com/pal-robotics/brax_training_viewer