Transformers
PyTorch
megatron-bert