将 Transformer 引擎支持从 1.12 升级到 2.18

目前 GPT-NeoX 将 NVIDIA Transformer 引擎固定为 `Transformer-engine[pytorch]==1.12`。最新稳定的 Transformer 引擎版本是 2.18.0。