add flax model

config.json

@@ -16,6 +16,7 @@
   "num_attention_heads": 2,
   "num_hidden_layers": 2,
   "pad_token_id": 0,
+  "position_embedding_type": "absolute",
   "type_vocab_size": 2,
   "vocab_size": 32000
 }