11 dependents
Package Description Downloads/month
Tensors and Dynamic neural networks in Python with strong GPU acceleration 83.9M
RAFT contains fundamental widely-used algorithms and primitives for machine lear... 21K
TensorRT LLM provides users with an easy-to-use Python API to define Large Langu... 16K
MLX: An array framework for Apple silicon 15K
RAFT contains fundamental widely-used algorithms and primitives for machine lear... 2K
cuVS - a library for vector search and clustering on the GPU 1K
Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library... 806
cuSOLVERMp - runtime libraries 617
wholegraph 421
cuBLASMp - runtime libraries 372
NCCL device-initiated API for the CuTe Python DSL 110