allow flax

commit from yhk

First version of the your-model-name model and tokenizer.

Update from yhk

initial commit