In general pytorch doesn’t really care what python structures you use to store your data.  The main caveat here is that when you create the dataloader you may need to specify a collate_fn which takes the individual samples and combines them into a batch.
See https://pytorch.org/docs/stable/data.html#dataloader-collate-fn for more details on how you define that.
Can write up a simple example later if that isn’t clear.