NVIDIA / cuda-python
CUDA Python: Performance meets Productivity
☆1,045Updated this week
Alternatives and similar repositories for cuda-python:
Users that are interested in cuda-python are comparing it to the libraries listed below
- CUDA Library Samples☆1,720Updated 3 weeks ago
- An Aspiring Drop-In Replacement for NumPy at Scale☆819Updated last week
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆330Updated 3 weeks ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,016Updated 9 months ago
- CUDA Core Compute Libraries☆1,385Updated this week
- common in-memory tensor structure☆930Updated 3 months ago
- CUDA Kernel Benchmarking Library☆547Updated last month
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆587Updated 2 months ago
- CUDA integration for Python, plus shiny features☆1,896Updated 2 months ago
- An open-source efficient deep learning framework/compiler, written in python.☆668Updated this week
- A GPU performance profiling tool for PyTorch models☆500Updated 3 years ago
- PyTriton is a Flask/FastAPI-like interface that simplifies Triton's deployment in Python environments.☆763Updated last month
- Pipeline Parallelism for PyTorch☆736Updated 4 months ago
- RAPIDS Memory Manager☆526Updated this week
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,239Updated this week
- Learn CUDA Programming, published by Packt☆1,073Updated last year
- Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)☆658Updated 4 months ago
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,690Updated last year
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆2,086Updated this week
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆755Updated last week
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆484Updated 3 weeks ago
- TensorRT Model Optimizer is a unified library of state-of-the-art model optimization techniques such as quantization, pruning, distillati…☆667Updated last week
- Triton Model Analyzer is a CLI tool to help with better understanding of the compute and memory requirements of the Triton Inference Serv…☆446Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,403Updated this week
- RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…☆816Updated this week
- NVIDIA Math Libraries for the Python Ecosystem☆220Updated 3 weeks ago
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,401Updated this week
- Reference implementations of MLPerf™ inference benchmarks☆1,274Updated this week
- ☆503Updated this week
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆895Updated this week