pytorch version of https://github.com/ZhuiyiTechnology/simbert