Real-time Web Viewer

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](https://summerofcode.withgoogle.com/programs/2025/projects/Xm0toJHl). This work aims to address the long-standing request for a live visualization feature, as discussed in issues **[#47](https://github.com/google/brax/issues/47)** and **[#107](https://github.com/google/brax/issues/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