refactor: reorganize dataset into lm/vlm/vam subdirectories

Add tokenizer training script and save trained tokenizer into checkpoint dir

Flatten src layout: drop omni namespace, modules live directly under src/