PyTorch2.1
torch_npu
torch_npu.contrib
torch_npu.jit
torch_npu.npu
torch_npu.npu.amp
torch_npu.optim
torch_npu.profiler
C++ API
Distributed
父主题：
PyTorch Ascend Adapter自定义API