refactoring dataloader into registries. (#1165)
* refactoring dataloader into registries
* lint
* fixing get_tokens_per_batch_func import issues
* lint
* resolving comments
* resolving comments
---------
Co-authored-by: Daniel King <43149077+dakinggg@users.noreply.github.com>
