upload flax model

allow flax

First version of the SSPT for bert-large-uncased model and tokenizer

initial commit