Torch MUSA

基于 PyTorch 扩展的 Python 包，用户可以通过 Torch-MUSA 充分发挥摩尔线程全功能 GPU 的强大性能。Torch-MUSA 实现了对 CUDA 的兼容，大幅降低了适配新算子的工作量，简化了开发流程。Torch-MUSA API 在格式上与 PyTorch 一致，让熟悉 PyTorch 的用户可以平滑迁移到 Torch-MUSA。Torch-MUSA 还提供了一组工具包，助力用户轻松进行 CUDA 移植、构建 MUSA 扩展和调试。