GPU Optimization for Python with NVIDIA
Read the specific reference before writing code — they contain detailed API patterns, optimization techniques, and pitfalls specific to each library.
LLM Evaluation
Evaluated by: xiaomi/mimo-v2-flash:free
Last evaluated: May 17, 2026
Prompt Preview
---
name: optimize-for-gpu
description: "GPU-accelerate Python code using CuPy, Numba CUDA, Warp, cuDF, cuML, cuGraph, KvikIO, cuCIM, cuxfilter, cuVS, cuSpatial, and RAFT. Use whenever the user mentions GPU/CUDA/NVIDIA acceleration, or wants to speed up NumPy, pandas, scikit-learn, scikit-image, NetworkX, GeoPandas, or Faiss workloads. Covers physics simulation, differentiable rendering, mesh ray casting, particle systems (DEM/SPH/fluids), vector/similarity search, GPUDirect Storage file IO, int...Full prompt length: 34150 characters
Tools & Technologies
- Python
- python