pip install git+https://github.com/NVIDIA/TransformerEngine.git@stable