Following my comment on #36 (closed), I think there are a lot of classes in this package would be more useful if they were just functions. Especially normalizers, data augmentations, networks (should be called architectures IMHO), and losses.

What do you think @tiago.pereira