Separate code for better maintainability

[ebursztein]

Keep only what is need for inference under the tf/
mode training code / layer and utils under training/
create a tokenizer class (we can do this after) that is for when not using layers. Move the code for this from layer to that class.