Adding `safetensors` variant of this model

updated for paper publication

upload flax model

allow flax

adding model card

initial version

initial commit