upload flax model

allow flax

Extended explanation in README.

Include copy+paste code for use.

Include slight sectioning.

Created README pointing to our paper and purpose of the model.

Adding other files that should go in here.

Adding initial weights for best-performing CLS run.

initial commit