upload flax model

allow flax

New version (distilbert-base to bert-base; attention matrices imitation)

First version

initial commit