Ascend PyTorch adapter
本项目开发了名为npu_inductor_mlir(Integrating PyTorch with MLIR for NPU acceleration.)的扩展库,支持用户基于PyTorch框架和torch_npu插件在昇腾NPU上使用图模式进行训练和推理。
Torch inductor implementations for NPU
BMTrain 是一个高效的大模型训练工具包,可以用于训练数百亿参数的大模型。BMTrain 可以在分布式训练模型的同时,能够保持代码的简洁性。