16 dependents
Package Description Downloads/month
CUDA Python: Performance meets Productivity 40.3M
NumPy & SciPy for GPU 3.4M
CUDA Python: Performance meets Productivity 1.5M
Python bindings for NVSHMEM 794K
The CUDA target for Numba 468K
NumPy & SciPy for GPU 340K
Python bindings for NVSHMEM 101K
NVIDIA Math Python libraries 46K
NumPy & SciPy for GPU 42K
Optimized primitives for collective multi-GPU communication 32K
CUDA Core Compute Libraries 17K
4K
Pytorch routines for (Ker)nel (Mac)hines 642
Package for converting TensorFlow DenseNet121 and DenseNet201 models to ONNX and... 415
A utterly useless package that imports everything for you. Now with top 1000 PyP... 255
NCCL device-initiated API for the CuTe Python DSL 120