add a new flax example for Bert model inference (#34794)

add a new flax example for Bert model inference (#34794) #6748