Plate Notation and Datasets in TensorFlow Probability