Ascend/pytorch
Ascend PyTorch adapter (torch_npu). Mirror of https://gitcode.com/Ascend/pytorch
This project provides an extension for PyTorch, called torch_npu, to enable your PyTorch-based AI models and workflows to run on Ascend NPU hardware. It takes your existing PyTorch code and allows it to leverage the high-performance computing capabilities of Ascend AI Processors. AI/ML engineers and researchers who are developing and deploying deep learning models will find this useful for accelerating their computational tasks.
493 stars.
Use this if you are a machine learning engineer or researcher using PyTorch and want to significantly speed up your deep learning model training and inference by utilizing Ascend AI Processors.
Not ideal if you are not working with PyTorch, do not have access to Ascend AI Processors, or your application does not require high-performance AI acceleration.
Stars
493
Forks
48
Language
Python
License
—
Category
Last pushed
Mar 13, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/Ascend/pytorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras-team/keras
Deep Learning for humans
Lightning-AI/torchmetrics
Machine learning metrics for distributed, scalable PyTorch applications.
Lightning-AI/pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
lanpa/tensorboardX
tensorboard for pytorch (and chainer, mxnet, numpy, ...)