[data.augmentations] Move transforms to augmentations; Implement auto-parellisation for ElasticDeformation (partial affects #5); Reflect changes in all relevant submodules

[data.transforms] Properly implement ElasticDeformation so it works with images or batches; Update test case for transforms; Create new test case for image_utils; Update models to reflect changes (partially addresses #5)

[datamodule] Slightly streamlines the datamodule approach; adds documentation; adds type annotations; adds TODOs