TPU/GPU visualization for learning from pixels
Open
enhancement
- Dominant language
- Jupyter Notebook
- Stars
- 3.2k
- Forks
- 349
- PR merge metrics
- No merged PRs in 30d
Description
Brax's rendering is CPU-only at the moment. For agents with vision, it would be useful for rendering to happen on accelerator so that training can stay fast.
[Vispy](https://vispy.org/) may be useful, but a fast native jax renderer would be even better for data on device compatability.
Contributor guide
Assessment
This issue has not been assessed yet.