CUDA integration for Python, plus shiny features
☆2,030Feb 22, 2026Updated 3 weeks ago
Alternatives and similar repositories for pycuda
Users that are interested in pycuda are comparing it to the libraries listed below
Sorting:
- OpenCL integration for Python, plus shiny features☆1,130Mar 9, 2026Updated last week
- Python interface to GPU-powered libraries☆989Oct 15, 2023Updated 2 years ago
- NumPy & SciPy for GPU☆10,847Updated this week
- NumPy aware dynamic Python compiler using LLVM☆10,935Updated this week
- Python module for performing basic dense linear algebra computations on the GPU using CUDA.☆610May 28, 2020Updated 5 years ago
- A code generator for array-based code on CPUs and GPUs☆626Mar 12, 2026Updated last week
- CUDA Python: Performance meets Productivity☆3,185Updated this week
- Open standard for machine learning interoperability☆20,484Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,800Mar 9, 2026Updated last week
- Seamless operability between C++11 and Python☆17,757Mar 7, 2026Updated last week
- Development repository for the Triton language and compiler☆18,656Updated this week
- [ARCHIVED] The C++ parallel algorithms library. See https://github.com/NVIDIA/cccl☆5,000Feb 8, 2024Updated 2 years ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,642Updated this week
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,821Oct 9, 2023Updated 2 years ago
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,442Updated this week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆35,108Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,931Mar 10, 2026Updated last week
- Open Machine Learning Compiler Framework☆13,197Updated this week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,958Updated this week
- An easy to use PyTorch to TensorRT converter☆4,858Aug 17, 2024Updated last year
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,679Dec 1, 2025Updated 3 months ago
- cuDF - GPU DataFrame Library☆9,558Updated this week
- Build and run Docker containers leveraging NVIDIA GPUs☆17,511Dec 6, 2023Updated 2 years ago
- ArrayFire: a general purpose GPU library.☆4,868Mar 7, 2026Updated last week
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆10,426Mar 13, 2026Updated last week
- Samples for CUDA Developers which demonstrates features in CUDA Toolkit☆8,953Jan 6, 2026Updated 2 months ago
- ONNX-TensorRT: TensorRT backend for ONNX☆3,194Feb 3, 2026Updated last month
- Optimized primitives for collective multi-GPU communication☆4,531Updated this week
- Simplify your onnx model☆4,309Feb 26, 2026Updated 3 weeks ago
- [ARCHIVED] The C++ Standard Library for your entire system. See https://github.com/NVIDIA/cccl☆2,308Feb 7, 2024Updated 2 years ago
- The most widely used Python to C compiler☆10,652Updated this week
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆98,243Updated this week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,926Mar 10, 2026Updated last week
- Python wrappers for the NVIDIA cuDNN libraries☆142May 26, 2017Updated 8 years ago
- Efficiently computes derivatives of NumPy code.☆7,464Updated this week
- Pure Python GPGPU library☆170Sep 17, 2025Updated 6 months ago
- Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.☆41,773Mar 13, 2026Updated last week
- Pyculib - Python bindings for CUDA libraries☆99Feb 18, 2019Updated 7 years ago
- oneAPI Deep Neural Network Library (oneDNN)☆3,964Updated this week