CUDA integration for Python, plus shiny features
☆2,025Feb 18, 2026Updated last week
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,131Updated this week
- Python interface to GPU-powered libraries☆989Oct 15, 2023Updated 2 years ago
- NumPy & SciPy for GPU☆10,804Updated this week
- NumPy aware dynamic Python compiler using LLVM☆10,921Feb 20, 2026Updated last week
- A code generator for array-based code on CPUs and GPUs☆626Updated this week
- Open standard for machine learning interoperability☆20,373Updated this week
- CUDA Python: Performance meets Productivity☆3,173Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,723Updated this week
- Seamless operability between C++11 and Python☆17,726Feb 17, 2026Updated last week
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,634Feb 19, 2026Updated last week
- Python module for performing basic dense linear algebra computations on the GPU using CUDA.☆610May 28, 2020Updated 5 years ago
- Development repository for the Triton language and compiler☆18,460Updated this week
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,315Updated this week
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,820Oct 9, 2023Updated 2 years ago
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,940Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,926Feb 16, 2026Updated last week
- [ARCHIVED] The C++ parallel algorithms library. See https://github.com/NVIDIA/cccl☆4,998Feb 8, 2024Updated 2 years ago
- Open Machine Learning Compiler Framework☆13,142Updated this week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,947Updated this week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,675Dec 1, 2025Updated 2 months ago
- cuDF - GPU DataFrame Library☆9,498Updated this week
- ArrayFire: a general purpose GPU library.☆4,859Sep 5, 2025Updated 5 months ago
- An easy to use PyTorch to TensorRT converter☆4,857Aug 17, 2024Updated last year
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆10,375Feb 21, 2026Updated last week
- Build and run Docker containers leveraging NVIDIA GPUs☆17,498Dec 6, 2023Updated 2 years ago
- Optimized primitives for collective multi-GPU communication☆4,474Updated this week
- ONNX-TensorRT: TensorRT backend for ONNX☆3,188Feb 3, 2026Updated 3 weeks ago
- Samples for CUDA Developers which demonstrates features in CUDA Toolkit☆8,870Jan 6, 2026Updated last month
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,860Updated this week
- Efficiently computes derivatives of NumPy code.☆7,460Updated this week
- [ARCHIVED] The C++ Standard Library for your entire system. See https://github.com/NVIDIA/cccl☆2,308Feb 7, 2024Updated 2 years ago
- Simplify your onnx model☆4,297Updated this week
- oneAPI Deep Neural Network Library (oneDNN)☆3,956Updated this week
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆97,688Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.☆41,413Feb 21, 2026Updated last week
- Compiler for Neural Network hardware accelerators☆3,324May 11, 2024Updated last year
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆19,389Updated this week
- The most widely used Python to C compiler☆10,627Updated this week