RLDS

RLDS (Reinforcement Learning Datasets) is a TensorFlow-Datasets format for episodic robot and RL data. It is the format behind the cross-embodiment Open X-Embodiment / RT-X collection, making it a natural choice when training on or alongside that data.