torch_waymo

Python library for loading the Waymo Open Dataset in PyTorch

By William Guimont-Martin • 1 minute read •


Here’s a small library I made to use the Waymo Open Dataset in PyTorch, without any dependency to Tensorflow: willGuimont/torch_waymo.

Here’s the corresponding PyPI package: torch-waymo.