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.