OpPlugin of Ascend Extension for PyTorch
TorchAir 支持用户基于PyTorch框架和torch_npu插件在昇腾NPU上使用图模式进行训练和推理
TorchAir 支持用户基于PyTorch框架和torch_npu插件在昇腾NPU上使用图模式进行训练和推理
OpPlugin of Ascend Extension for PyTorch 原commit id: 6f44f4f5d777fbe3370e913853e29723177d8a25
BMTrain 是一个高效的大模型训练工具包,可以用于训练数百亿参数的大模型。BMTrain 可以在分布式训练模型的同时,能够保持代码的简洁性。