pytorch jax bert masked-lm transformers fill-mask
.gitattributes config.json flax_model.msgpack merges.txt pytorch_model.bin training_args.bin vocab.json checkpoint-17000/config.json checkpoint-17000/optimizer.pt checkpoint-17000/pytorch_model.bin checkpoint-17000/scheduler.pt checkpoint-17000/trainer_state.json checkpoint-17000/training_args.bin