[FEATURE] Dataset class registry

[sdatkinson]

companion to #334, I want the ability to extend the data set classes and use them in the trainer without needing to modify this package's code.

To do this, I'm going to make a registry similar to the model registry enabling people to add in new data set classes so that they can extend this package's framework to easily try new things.