Adding `safetensors` variant of this model (#1)

upload flax model

allow flax

add widget example

Create README.md

Train model to loss 0.8

Initial commit

initial commit