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
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
How to Test
We invite the community to test this prototype. You can get started here:
https://github.com/pal-robotics/brax_training_viewer