NVIDIA / cuda-python
CUDA Python Low-level Bindings
☆984Updated this week
Alternatives and similar repositories for cuda-python:
Users that are interested in cuda-python are comparing it to the libraries listed below
- An Aspiring Drop-In Replacement for NumPy at Scale☆758Updated 2 weeks ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆312Updated last week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,014Updated 7 months ago
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆566Updated last month
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆458Updated last month
- NVIDIA Math Libraries for the Python Ecosystem☆212Updated 2 weeks ago
- common in-memory tensor structure☆911Updated 2 months ago
- CUDA Core Compute Libraries☆1,319Updated this week
- CUDA Kernel Benchmarking Library☆526Updated last week
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆742Updated last week
- RAPIDS Memory Manager☆494Updated this week
- Learn CUDA Programming, published by Packt☆1,036Updated 11 months ago
- CUDA integration for Python, plus shiny features☆1,859Updated 3 weeks ago
- CUDA Library Samples☆1,640Updated this week
- The Foundation for All Legate Libraries☆198Updated 2 weeks ago
- TensorRT Model Optimizer is a unified library of state-of-the-art model optimization techniques such as quantization, pruning, distillati…☆589Updated last week
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,221Updated last week
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆2,000Updated this week
- Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)☆620Updated 3 months ago
- An open-source efficient deep learning framework/compiler, written in python.☆656Updated 2 weeks ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,398Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆274Updated this week
- RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…☆789Updated this week
- Pipeline Parallelism for PyTorch☆729Updated 3 months ago
- PyTriton is a Flask/FastAPI-like interface that simplifies Triton's deployment in Python environments.☆746Updated last week
- ☆488Updated this week
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,686Updated last year
- A GPU performance profiling tool for PyTorch models☆495Updated 3 years ago
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆879Updated last week
- Hands-On GPU Programming with Python and CUDA, published by Packt☆351Updated 3 months ago