cupy

GPU computing library

A Python library for running NumPy/SciPy code on NVIDIA CUDA or AMD ROCm platforms using GPU acceleration.

NumPy & SciPy for GPU

GitHub

10k stars
127 watching
859 forks
Language: Python
last commit: over 1 year ago
Linked from 4 awesome lists

cublascudacudnncupycurandcusolvercusparsecusparseltcutensorgpuncclnumpynvrtcnvtxpythonrocmscipytensor

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
pyqtgraph/pyqtgraph A high-performance data visualization and GUI library for scientific applications. 3,924
numpy/numpy A comprehensive library providing efficient numerical computation and data manipulation capabilities for Python-based scientific computing. 28,350
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 84,978
wookayin/gpustat Monitors and displays information about NVIDIA graphics devices 4,094
xuehaipan/nvitop An interactive tool for monitoring and managing NVIDIA GPUs, displaying device and process information in an intuitive interface. 4,921
rapidsai/cuml A suite of libraries implementing machine learning algorithms and mathematical primitives on NVIDIA GPUs 4,292
neka-nat/cupoch A C++ library for rapid 3D data processing and robotics computation using GPU computing. 933
pkmital/tensorflow_tutorials A collection of examples and tutorials illustrating various applications and use cases of TensorFlow in Python 5,636
rapidsai/cudf A GPU-accelerated data manipulation library built on top of C++/CUDA and Apache Arrow. 8,534
turboderp/exllama A re-implementation of Llama for efficient use with quantized weights on modern GPUs. 2,783
conda-forge/miniforge Pre-configured package installer with support for various CPU architectures and Python implementations 6,728
cornellius-gp/gpytorch A library for creating scalable and flexible Gaussian process models with ease 3,605
sympy/sympy A computer algebra system written in Python to solve mathematical equations and expressions 13,130
sniklaus/pytorch-extension A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors. 117
graphistry/pygraphistry A Python library for fast and efficient graph data processing, visualization, and analysis on GPUs 2,184