GitHub - Peterc3-dev/torch-vulkan: Vulkan compute backend for PyTorch — runs on any GPU. PrivateUse1 dispatch, SPIR-V shaders, zero ROCm/CUDA dependency.

Name		Name	Last commit message	Last commit date
Latest commit History 28 Commits
csrc		csrc
tests		tests
torch_vulkan		torch_vulkan
.gitignore		.gitignore
CMakeLists.txt		CMakeLists.txt
persistent_pipeline.py		persistent_pipeline.py
setup.py		setup.py

About

Vulkan compute backend for PyTorch — runs on any GPU. PrivateUse1 dispatch, SPIR-V shaders, zero ROCm/CUDA dependency.

No releases published