An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.

pypi.org "gpu" keyword

nvidia-cutlass 4.2.0.0
CUTLASS
9 versions - Latest release: about 1 year ago - 250 thousand downloads last month - 10,713 stars on GitHub - 1 maintainer
nvidia-cutlass-dsl-libs-cu12 4.8.1
NVIDIA CUTLASS Python DSL
9 versions - Latest release: 30 days ago - 4.24 million downloads last month - 10,713 stars on GitHub - 1 maintainer
nvidia-cutlass-dsl 4.8.1
NVIDIA CUTLASS Python DSL
34 versions - Latest release: 30 days ago - 5.08 million downloads last month - 10,713 stars on GitHub - 1 maintainer
tatolab-stream 0.69.20
The Python SDK for Tatolab streams. To get started, install the Tatolab app: see tatolab.com/docs...
126 versions - Latest release: 27 minutes ago - 12.2 thousand downloads last month - 6 stars on GitHub - 1 maintainer
jaxht 0.5.0
JAX-native spherical harmonic transforms (spin-0/2 on-grid + off-grid NUFFT): GPU-capable, differ...
10 versions - Latest release: about 2 hours ago - 179 downloads last month - 0 stars on GitHub - 1 maintainer
oceanctl 4.3.0
Ocean CLI — 작업 제출·상태 확인을 터미널에서
15 versions - Latest release: about 1 hour ago - 111 downloads last month - 1 maintainer
Top 4.4% on pypi.org
cupy-cuda80 7.8.0 💰
CuPy: NumPy-like API accelerated with CUDA
30 versions - Latest release: about 6 years ago - 5 dependent packages - 7 dependent repositories - 3.16 thousand downloads last month - 12,406 stars on GitHub - 7 maintainers
Top 1.8% on pypi.org
cupy-cuda111 12.3.0 💰
CuPy: NumPy & SciPy for GPU
30 versions - Latest release: almost 3 years ago - 12 dependent packages - 76 dependent repositories - 1.66 thousand downloads last month - 12,406 stars on GitHub - 3 maintainers
Top 2.3% on pypi.org
cupy-cuda112 10.6.0 💰
CuPy: NumPy & SciPy for GPU
17 versions - Latest release: over 4 years ago - 9 dependent packages - 22 dependent repositories - 1.8 thousand downloads last month - 12,406 stars on GitHub - 2 maintainers
Top 4.3% on pypi.org
arrayfire 3.8.0
Python bindings for ArrayFire
24 versions - Latest release: over 4 years ago - 5 dependent packages - 13 dependent repositories - 1.08 thousand downloads last month - 4,911 stars on GitHub - 3 maintainers
Top 0.3% on pypi.org
deepspeed 0.20.0
DeepSpeed library
136 versions - Latest release: 2 days ago - 109 dependent packages - 3,724 dependent repositories - 1.34 million downloads last month - 43,327 stars on GitHub - 3 maintainers
executorch 1.6.0
On-device AI across mobile, embedded and edge for PyTorch
20 versions - Latest release: 10 days ago - 68.3 thousand downloads last month - 5,168 stars on GitHub - 4 maintainers
Top 1.8% on pypi.org
cupy-cuda110 12.3.0 💰
CuPy: NumPy & SciPy for GPU
32 versions - Latest release: almost 3 years ago - 12 dependent packages - 36 dependent repositories - 1.59 thousand downloads last month - 12,406 stars on GitHub - 5 maintainers
Top 2.8% on pypi.org
cupy-cuda117 10.6.0 💰
CuPy: NumPy & SciPy for GPU
1 version - Latest release: over 4 years ago - 6 dependent packages - 10 dependent repositories - 144 downloads last month - 12,406 stars on GitHub - 2 maintainers
Top 2.5% on pypi.org
cupy-cuda101 9.6.0 💰
CuPy: NumPy & SciPy for GPU
33 versions - Latest release: about 5 years ago - 7 dependent packages - 107 dependent repositories - 1.58 thousand downloads last month - 12,406 stars on GitHub - 5 maintainers
Top 2.2% on pypi.org
cupy-cuda113 10.6.0 💰
CuPy: NumPy & SciPy for GPU
13 versions - Latest release: over 4 years ago - 6 dependent packages - 112 dependent repositories - 581 downloads last month - 12,406 stars on GitHub - 2 maintainers
Top 3.2% on pypi.org
cupy-cuda90 8.6.0 💰
CuPy: A NumPy-compatible array library accelerated by CUDA
37 versions - Latest release: over 5 years ago - 5 dependent packages - 21 dependent repositories - 880 downloads last month - 12,406 stars on GitHub - 7 maintainers
Top 1.8% on pypi.org
cupy-cuda100 9.6.0 💰
CuPy: NumPy & SciPy for GPU
36 versions - Latest release: about 5 years ago - 7 dependent packages - 205 dependent repositories - 949 downloads last month - 12,406 stars on GitHub - 5 maintainers
cupy-rocm-4-2 10.6.0 💰
CuPy: NumPy & SciPy for GPU
13 versions - Latest release: over 4 years ago - 1 dependent repositories - 89 downloads last month - 12,406 stars on GitHub - 2 maintainers
Top 4.1% on pypi.org
cupy-cuda91 7.8.0 💰
CuPy: NumPy-like API accelerated with CUDA
30 versions - Latest release: about 6 years ago - 5 dependent packages - 6 dependent repositories - 598 downloads last month - 12,406 stars on GitHub - 7 maintainers
Top 3.2% on pypi.org
cupy-cuda115 10.6.0 💰
CuPy: NumPy & SciPy for GPU
9 versions - Latest release: over 4 years ago - 4 dependent packages - 20 dependent repositories - 663 downloads last month - 12,406 stars on GitHub - 2 maintainers
cupy-cuda13x 14.2.0 💰
CuPy: NumPy & SciPy for GPU
6 versions - Latest release: 3 months ago - 584 thousand downloads last month - 12,406 stars on GitHub
Top 2.1% on pypi.org
cupy-cuda102 12.3.0 💰
CuPy: NumPy & SciPy for GPU
40 versions - Latest release: almost 3 years ago - 11 dependent packages - 58 dependent repositories - 3.14 thousand downloads last month - 12,406 stars on GitHub - 5 maintainers
Top 2.2% on pypi.org
cupy-cuda114 10.6.0 💰
CuPy: NumPy & SciPy for GPU
12 versions - Latest release: over 4 years ago - 5 dependent packages - 21 dependent repositories - 971 downloads last month - 12,406 stars on GitHub - 2 maintainers
cupy-rocm-4-0 10.6.0 💰
CuPy: NumPy & SciPy for GPU
15 versions - Latest release: over 4 years ago - 1 dependent repositories - 101 downloads last month - 12,406 stars on GitHub - 2 maintainers
Top 6.9% on pypi.org
cupy-rocm-5-0 13.3.0 💰
CuPy: NumPy & SciPy for GPU
21 versions - Latest release: about 2 years ago - 1 dependent package - 1 dependent repositories - 122 downloads last month - 12,406 stars on GitHub - 4 maintainers
Top 1.0% on pypi.org
cupy 14.2.0 💰
CuPy: NumPy & SciPy for GPU
152 versions - Latest release: 3 months ago - 91 dependent packages - 495 dependent repositories - 37.7 thousand downloads last month - 12,406 stars on GitHub - 10 maintainers
Top 2.4% on pypi.org
cupy-cuda12x 14.2.0 💰
CuPy: NumPy & SciPy for GPU
20 versions - Latest release: 3 months ago - 45 dependent packages - 4 dependent repositories - 1.45 million downloads last month - 12,406 stars on GitHub - 4 maintainers
cupy-rocm-7-0 14.2.0 💰
CuPy: NumPy & SciPy for GPU
5 versions - Latest release: 3 months ago - 3.89 thousand downloads last month - 12,406 stars on GitHub
Top 3.6% on pypi.org
cupy-cuda92 9.6.0 💰
CuPy: NumPy & SciPy for GPU
42 versions - Latest release: about 5 years ago - 5 dependent packages - 11 dependent repositories - 714 downloads last month - 12,406 stars on GitHub - 7 maintainers
Top 7.4% on pypi.org
cupy-rocm-4-3 13.3.0 💰
CuPy: NumPy & SciPy for GPU
27 versions - Latest release: about 2 years ago - 2 dependent packages - 1 dependent repositories - 173 downloads last month - 12,406 stars on GitHub - 3 maintainers
Top 1.2% on pypi.org
cupy-cuda11x 13.6.0 💰
CuPy: NumPy & SciPy for GPU
20 versions - Latest release: about 1 year ago - 44 dependent packages - 118 dependent repositories - 22.6 thousand downloads last month - 12,406 stars on GitHub - 5 maintainers
Top 2.9% on pypi.org
cupy-cuda116 10.6.0 💰
CuPy: NumPy & SciPy for GPU
5 versions - Latest release: over 4 years ago - 4 dependent packages - 20 dependent repositories - 896 downloads last month - 12,406 stars on GitHub - 2 maintainers
cuda-tile-preview 1.8.1
Preview APIs for cuda-tile
3 versions - Latest release: 2 days ago - 303 downloads last month - 2,168 stars on GitHub - 1 maintainer
cuda-tile 1.8.1
CUDA Tile Compiler
22 versions - Latest release: 2 days ago - 3.33 million downloads last month - 2,168 stars on GitHub - 1 maintainer
sqtop 0.5.0
htop for Slurm: partitions, account/QOS limits, estimated start times and your jobs, across clust...
6 versions - Latest release: about 3 hours ago - 563 downloads last month - 0 stars on GitHub - 1 maintainer
cuda-cooperative 0.0.1.dev0
cuda.cooperative: (experimental) CUDA cooperative algorithms for Python
1 version - Latest release: over 1 year ago - 20 downloads last month - 2,573 stars on GitHub - 4 maintainers
cuda-parallel 0.0.1.dev0
cuda.parallel: (experimental) CUDA parallel algorithms for Python
1 version - Latest release: over 1 year ago - 18 downloads last month - 2,573 stars on GitHub - 4 maintainers
cuda-cccl 1.3.0
CUDA Core Compute Libraries for Python
29 versions - Latest release: 18 days ago - 5.82 thousand downloads last month - 2,573 stars on GitHub - 4 maintainers
Top 2.6% on pypi.org
faiss-gpu 1.16.0
A library for efficient similarity search and clustering of dense vectors (GPU support).
17 versions - Latest release: 4 days ago - 51 dependent packages - 367 dependent repositories - 91.1 thousand downloads last month - 41,268 stars on GitHub - 3 maintainers
Top 1.1% on pypi.org
bitsandbytes 0.50.2 💰
k-bit optimizers and matrix multiplication routines.
67 versions - Latest release: 3 months ago - 240 dependent packages - 3,758 dependent repositories - 3.65 million downloads last month - 8,537 stars on GitHub - 2 maintainers
jax-hpc-profiler 0.3.4
A comprehensive benchmarking and profiling tool designed for JAX in HPC environments, offering au...
18 versions - Latest release: 4 months ago - 86 downloads last month - 3 stars on GitHub - 1 maintainer
infinity-institute 0.1.0
Python client for Infinity Media Server - GPU-accelerated image, video, and audio generation
1 version - Latest release: 10 months ago - 38 downloads last month - 1 maintainer
magnetron 0.2.3 💰
A compact machine learning runtime for developers who want to understand, control, and optimize t...
14 versions - Latest release: 14 days ago - 824 downloads last month - 709 stars on GitHub - 1 maintainer
Top 1.6% on pypi.org
nemo-toolkit 3.0.0
NeMo - a toolkit for Conversational AI
109 versions - Latest release: 3 months ago - 4 dependent packages - 27 dependent repositories - 448 thousand downloads last month - 18,727 stars on GitHub - 3 maintainers
Top 0.1% on pypi.org
torch 2.15.0
Tensors and Dynamic neural networks in Python with strong GPU acceleration
55 versions - Latest release: 18 days ago - 6,944 dependent packages - 92,053 dependent repositories - 60.7 million downloads last month - 105,267 stars on GitHub - 4 maintainers
accelana-torch-runtime 2.14.1
Accelana's repackaged PyTorch CPU runtime for Windows ARM64
1 version - Latest release: about 1 month ago - 28 downloads last month - 105,267 stars on GitHub - 1 maintainer
pytorch-riscv64 2.3.0a1
Tensors and Dynamic neural networks in Python
2 versions - Latest release: over 2 years ago - 16 downloads last month - 105,267 stars on GitHub - 1 maintainer
nvidia-cudnn-frontend 1.32.3
NVIDIA cuDNN Frontend — Python and C++ Graph API with SOTA attention (SDPA / Flash Attention), Mo...
41 versions - Latest release: 4 days ago - 4.21 million downloads last month - 998 stars on GitHub - 1 maintainer
Top 8.1% on pypi.org
taichi-nightly 0.5.11
The Taichi Programming Language
86 versions - Latest release: over 6 years ago - 2 dependent packages - 4 dependent repositories - 1.42 thousand downloads last month - 28,454 stars on GitHub - 1 maintainer
Top 1.4% on pypi.org
taichi 1.7.4
The Taichi Programming Language
56 versions - Latest release: over 1 year ago - 17 dependent packages - 113 dependent repositories - 178 thousand downloads last month - 28,454 stars on GitHub - 2 maintainers
fftlog-lss 0.1.3
JAX-accelerated FFTLog implementation for Large Scale Structure cosmology
4 versions - Latest release: about 6 hours ago - 748 downloads last month - 2 stars on GitHub - 2 maintainers
anuga 4.2.0
A set of python modules for tsunami and flood modelling
32 versions - Latest release: 1 day ago - 2 dependent repositories - 5.78 thousand downloads last month - 64 stars on GitHub - 1 maintainer
inferencekit-cli 0.1.1
Make your model serve faster on your GPU, and prove it. Skills for Claude Code, Codex and OpenCod...
2 versions - Latest release: 8 days ago - 470 downloads last month - 31 stars on GitHub - 1 maintainer
uc-manager-cann910-a3 0.9.1
Persist and reuse KV Cache to speedup your LLM.
7 versions - Latest release: 3 days ago - 399 downloads last month - 355 stars on GitHub - 1 maintainer
swm-gpu 0.3.16
One CLI to search, provision, and manage cloud GPUs across 10 providers
50 versions - Latest release: about 5 hours ago - 391 downloads last month - 27 stars on GitHub - 1 maintainer
uc-manager-cann900-a2 0.9.1
Persist and reuse KV Cache to speedup your LLM.
4 versions - Latest release: 3 days ago - 383 downloads last month - 355 stars on GitHub - 1 maintainer
uc-manager-cann921-a3 0.9.1
Persist and reuse KV Cache to speedup your LLM.
2 versions - Latest release: 3 days ago - 224 downloads last month - 355 stars on GitHub - 1 maintainer
uc-manager-cann900-a3 0.9.1
Persist and reuse KV Cache to speedup your LLM.
4 versions - Latest release: 3 days ago - 376 downloads last month - 355 stars on GitHub - 1 maintainer
uc-manager-cann921-a5 0.9.1
Persist and reuse KV Cache to speedup your LLM.
2 versions - Latest release: 3 days ago - 225 downloads last month - 355 stars on GitHub - 1 maintainer
narwhal-inference 0.9.0
Adaptive hot-swap disaggregation for LLM inference
11 versions - Latest release: about 8 hours ago - 268 downloads last month - 8 stars on GitHub - 1 maintainer
uc-manager-cuda-cu129 0.9.1
Persist and reuse KV Cache to speedup your LLM.
9 versions - Latest release: 3 days ago - 600 downloads last month - 355 stars on GitHub - 1 maintainer
uc-manager-cuda-cu130 0.9.1
Persist and reuse KV Cache to speedup your LLM.
9 versions - Latest release: 3 days ago - 601 downloads last month - 355 stars on GitHub - 1 maintainer
uc-manager-cann910-a2 0.9.1
Persist and reuse KV Cache to speedup your LLM.
7 versions - Latest release: 3 days ago - 424 downloads last month - 355 stars on GitHub - 1 maintainer
chitu 0.1.2
A high-performance inference framework for large language models, focusing on efficiency, flexibi...
3 versions - Latest release: over 1 year ago - 32 downloads last month - 3,000 stars on GitHub - 1 maintainer
uc-manager-cann910-a5 0.9.1
Persist and reuse KV Cache to speedup your LLM.
3 versions - Latest release: 3 days ago - 240 downloads last month - 355 stars on GitHub - 1 maintainer
uc-manager-cann921-a2 0.9.1
Persist and reuse KV Cache to speedup your LLM.
2 versions - Latest release: 3 days ago - 248 downloads last month - 355 stars on GitHub - 1 maintainer
threecrate-mcp 0.2.2
MCP server that gives AI agents threecrate's 3D point cloud and mesh tools
4 versions - Latest release: 9 days ago - 569 downloads last month - 52 stars on GitHub - 1 maintainer
Top 1.2% on pypi.org
chainer 7.8.1
A flexible framework of neural networks
111 versions - Latest release: almost 5 years ago - 10 dependent packages - 1,022 dependent repositories - 23.5 thousand downloads last month - 5,916 stars on GitHub - 8 maintainers
threecrate 0.12.1
Python bindings for threecrate: a high-performance 3D point cloud and mesh processing library
8 versions - Latest release: 15 days ago - 1.13 thousand downloads last month - 52 stars on GitHub - 1 maintainer
jittor-offline 0.0.7
jittor project
6 versions - Latest release: over 4 years ago - 11 downloads last month - 3,242 stars on GitHub - 1 maintainer
Top 4.1% on pypi.org
jittor 1.3.11.0
a Just-in-time(JIT) deep learning framework
405 versions - Latest release: 6 months ago - 1 dependent package - 45 dependent repositories - 1.87 thousand downloads last month - 3,242 stars on GitHub - 1 maintainer
meshoptimizer 0.1.0
Python wrapper for meshoptimizer library
7 versions - Latest release: over 1 year ago - 83.3 thousand downloads last month - 8,819 stars on GitHub - 1 maintainer
basalt-sass 1.0.0 💰
The check NVIDIA never shipped: an assembler, hazard checker and scheduler for consumer Blackwell...
1 version - Latest release: 3 months ago - 275 downloads last month - 15 stars on GitHub - 1 maintainer
subhkl 1.5.6
Solving crystal orientation from Laue images
10 versions - Latest release: about 6 hours ago - 542 downloads last month - 1 stars on GitHub - 1 maintainer
lemonade-sdk 9.1.4
Lemonade SDK: Your LLM Aide for Validation and Deployment
40 versions - Latest release: 10 months ago - 3.11 thousand downloads last month - 6,211 stars on GitHub - 1 maintainer
apache-tvm-tirx 0.1.0
TVM TIRx
12 versions - Latest release: 1 day ago - 2.77 thousand downloads last month - 13,899 stars on GitHub - 1 maintainer
hhb-tvm 3.2.3
TVM: An End to End Tensor IR/DSL Stack for Deep Learning Systems
11 versions - Latest release: over 1 year ago - 1 dependent package - 220 downloads last month - 13,899 stars on GitHub - 1 maintainer
Top 5.6% on pypi.org
apache-tvm 0.27.0.post1
Apache TVM: An End-to-End Deep Learning Compiler Stack
184 versions - Latest release: about 2 months ago - 2 dependent packages - 3 dependent repositories - 19.5 thousand downloads last month - 13,899 stars on GitHub - 6 maintainers
tvm-ffi
Machine learning for systems
1 version - Latest release: over 1 year ago - 99 downloads last month - 13,899 stars on GitHub - 1 maintainer
sdnn-cl 2.2.0
SDNN: An End to End Tensor IR/DSL Stack for Deep Learning Systems
1 version - Latest release: over 4 years ago - 2.33 thousand downloads last month - 13,899 stars on GitHub - 1 maintainer
apache-tvm-cu113 0.1.0
Open Machine Learning Compiler Framework
1 version - Latest release: over 2 years ago - 30 downloads last month - 13,899 stars on GitHub - 1 maintainer
xtc-tvm-python-bindings 0.21.0.8
TVM: An End to End Tensor IR/DSL Stack for Deep Learning Systems
9 versions - Latest release: 3 months ago - 538 downloads last month - 13,899 stars on GitHub - 1 maintainer
quantfit 0.61.0
Quantize an LLM and check it still refuses what it should — a GPU-aware quantization CLI that mea...
91 versions - Latest release: about 7 hours ago - 3.93 thousand downloads last month - 1 stars on GitHub - 1 maintainer
Top 3.5% on pypi.org
jetson-stats 7.3.0 💰
Interactive system-monitor and process viewer for all NVIDIA Jetson [Thor, Orin, Xavier, Nano, TX...
97 versions - Latest release: about 7 hours ago - 1 dependent package - 8 dependent repositories - 38.9 thousand downloads last month - 2,634 stars on GitHub - 1 maintainer
Top 1.8% on pypi.org
scalene 2.4.0 💰
Scalene: A high-resolution, low-overhead CPU, GPU, and memory profiler for Python with AI-powered...
191 versions - Latest release: 22 days ago - 5 dependent packages - 58 dependent repositories - 315 thousand downloads last month - 13,559 stars on GitHub - 1 maintainer
zamba 2.7.2
Zamba is a command line tool and Python package to identify animals in camera trap videos and tra...
7 versions - Latest release: 5 months ago - 1 dependent repositories - 51 downloads last month - 163 stars on GitHub - 1 maintainer
Top 6.9% on pypi.org
dpctl 0.23.0
A lightweight Python wrapper for a subset of SYCL.
20 versions - Latest release: 17 days ago - 2 dependent packages - 6 dependent repositories - 9.66 thousand downloads last month - 130 stars on GitHub - 2 maintainers
tacuda 0.9.0
CUDA technical analysis for Python: TA-Lib baseline indicators, fused portfolio batches, CUDA Gra...
3 versions - Latest release: about 9 hours ago - 145 downloads last month - 4 stars on GitHub - 1 maintainer
veros 1.7.0
The versatile ocean simulator, in pure Python, powered by JAX.
24 versions - Latest release: about 1 month ago - 4 dependent repositories - 289 downloads last month - 402 stars on GitHub - 2 maintainers
tensorcircuit-nightly 0.2.2
High performance unified quantum computing framework for the NISQ era
1,405 versions - Latest release: over 4 years ago - 1 dependent repositories - 18.8 thousand downloads last month - 95 stars on GitHub - 1 maintainer
pyclblast 1.4.0 💰
Python bindings for CLBlast, the tuned OpenCL BLAS library
6 versions - Latest release: over 2 years ago - 1 dependent repositories - 40 downloads last month - 1,195 stars on GitHub - 1 maintainer
Top 0.4% on pypi.org
fastai 2.8.12
fastai simplifies training fast and accurate neural nets using modern best practices
154 versions - Latest release: 2 months ago - 64 dependent packages - 2,806 dependent repositories - 236 thousand downloads last month - 28,268 stars on GitHub - 2 maintainers
cuda-bench 0.3.0
CUDA Kernel Benchmarking Package with CUDA 12.x and 13.x runtime support
4 versions - Latest release: 3 months ago - 4.21 thousand downloads last month - 945 stars on GitHub - 4 maintainers
openlit 1.45.0 💰
OpenTelemetry-native Auto instrumentation library for monitoring LLM Applications and GPUs, facil...
175 versions - Latest release: 3 months ago - 393 thousand downloads last month - 3,035 stars on GitHub - 1 maintainer
rocm-bindings-compiler 10.0.0.0.1.2
Python bindings for ROCm compiler infrastructure (LLVM, Comgr, Clang)
3 versions - Latest release: about 2 months ago - 536 downloads last month - 42 stars on GitHub - 1 maintainer
rocm-bindings-hip 10.0.0.0.1.2
HIP and HIPRTC Python bindings for ROCm
3 versions - Latest release: about 2 months ago - 55.1 thousand downloads last month - 42 stars on GitHub - 1 maintainer
mlflow-torchserve 0.1.0
Torch Serve Mlflow Deployment
3 versions - Latest release: about 6 years ago - 1 dependent repositories - 74 downloads last month - 4,139 stars on GitHub - 2 maintainers
nccl4py 0.6.0
NCCL4Py: Python bindings for NCCL
7 versions - Latest release: about 2 months ago - 2.48 million downloads last month - 5,227 stars on GitHub - 1 maintainer
roger 3.0.9
Runoff Generation Research - a process-based hydrological toolbox model in Python
10 versions - Latest release: over 1 year ago - 108 downloads last month - 22 stars on GitHub - 1 maintainer