Ascend Extension for PyTorch
自定义API
torch_npu
torch_npu.contrib
torch_npu.jit
torch_npu.npu
torch_npu.npu.amp
torch_npu.optim
torch_npu.profiler
torch_npu.utils
C++ API
Distributed
父主题：
PyTorch2.1.0