upload flax model

allow flax

First version of the neural magic BERT-squad-12 layer

First commit

initial commit