NVIDIA / cuda-python

Related projects ⓘ

Alternatives and complementary repositories for cuda-python

• nv-legate / cunumeric
  An Aspiring Drop-In Replacement for NumPy at Scale
  ☆620Updated last month
• dmlc / dlpack
  common in-memory tensor structure
  ☆905Updated last month
• NVIDIA / NVTX
  The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…
  ☆301Updated this week
• NVIDIA / multi-gpu-programming-models
  Examples demonstrating available options to program multiple GPUs in a single node or a cluster
  ☆553Updated last week
• NVIDIA / nvmath-python
  NVIDIA Math Libraries for the Python Ecosystem
  ☆202Updated 4 months ago
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆512Updated 2 weeks ago
• nv-legate / legate.core
  The Foundation for All Legate Libraries
  ☆189Updated last month
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,010Updated 6 months ago
• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆766Updated this week
• NVIDIA / cccl
  CUDA Core Compute Libraries
  ☆1,246Updated this week
• inducer / pycuda
  CUDA integration for Python, plus shiny features
  ☆1,849Updated this week
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,679Updated last year
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆492Updated this week
• NVIDIA / cuCollections
  ☆482Updated this week
• NVIDIA / CUDALibrarySamples
  CUDA Library Samples
  ☆1,600Updated this week
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,211Updated this week
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆727Updated this week
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆493Updated 3 years ago
• triton-inference-server / pytriton
  PyTriton is a Flask/FastAPI-like interface that simplifies Triton's deployment in Python environments.
  ☆739Updated this week
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,395Updated this week
• NVIDIA / cudnn-frontend
  cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it
  ☆446Updated 2 weeks ago
• rapidsai / kvikio
  KvikIO - High Performance File IO
  ☆156Updated this week
• dionhaefner / pyhpc-benchmarks
  A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python
  ☆311Updated last month
• triton-inference-server / model_analyzer
  Triton Model Analyzer is a CLI tool to help with better understanding of the compute and memory requirements of the Triton Inference Serv…
  ☆426Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆649Updated this week
• NVIDIA / TensorRT-Model-Optimizer
  TensorRT Model Optimizer is a unified library of state-of-the-art model optimization techniques such as quantization, pruning, distillati…
  ☆534Updated this week
• NVIDIA / pyxis
  Container plugin for Slurm Workload Manager
  ☆287Updated this week
• pytorch / PiPPy
  Pipeline Parallelism for PyTorch
  ☆725Updated 2 months ago
• pytorch / TensorRT
  PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
  ☆2,585Updated this week
• ROCm / HIPIFY
  HIPIFY: Convert CUDA to Portable C++ Code
  ☆523Updated this week