← All projects Libraries Load and convert the Waymo Open Dataset for PyTorchtorch_waymo
torch_waymo is a Python library for loading the Waymo Open Dataset in PyTorch without requiring TensorFlow in the training pipeline. It converts Waymo records into either simplified frames or complete frame data, then exposes the result through a PyTorch-friendly dataset interface.
The package includes a torch-waymo-convert command-line tool and can be installed from PyPI. This keeps the one-time dataset conversion separate from model development and makes Waymo data easier to use in a standard PyTorch workflow.