Pre-built CUDA Python wheels for ML/3D packages that are painful to compile from source.
Package Index · Package Index v2 · Dashboard · Install Helper · Full Build Matrix
| Package | Source |
|---|---|
| sageattention | thu-ml/SageAttention |
| gsplat | nerfstudio-project/gsplat |
| nvdiffrast | NVlabs/nvdiffrast |
| pytorch3d | facebookresearch/pytorch3d |
| cumesh | JeffreyXiang/CuMesh |
| cubvh | ashawkey/cubvh |
| ovoxel | microsoft/TRELLIS.2 |
| flexgemm | JeffreyXiang/FlexGEMM |
| nvdiffrec_render | NVlabs/nvdiffrec |
| torch_generic_nms | ronghanghu/torch_generic_nms |
| cc_torch | ronghanghu/cc_torch |
| spconv | traveller59/spconv |
| flash_attn | Dao-AILab/flash-attention |
| torch_cluster | pyg-team/pytorch_cluster |
| torch_scatter | pyg-team/pytorch_scatter |
| torch_sparse | pyg-team/pytorch_sparse |
| natten | SHI-Labs/NATTEN |
Used by comfy-env to automatically resolve and install CUDA packages for ComfyUI custom nodes.
See packages/README.md for build config format.
MIT