রোবোটিক্স ও এজ AI

NVIDIA Warp: Python kernels for CPU and GPU simulation

NVIDIA/warp

Compiles typed Python functions into compute kernels with geometry, physics and differentiable programming capabilities for simulation and machine learning.

★ 7.1Kতারকা
⑂ 621ফর্ক
242খোলা সমস্যাগুলো
Pythonভাষা
Apache-2.0লাইসেন্স
Q92সম্পাদকীয় স্কোর

ওভারভিউ

Warp lets developers write parallel kernels in a Python-based language and just-in-time compile them for CPUs or supported NVIDIA GPUs. It provides vector and matrix types, geometry operations and simulation components, with differentiable computation that can participate in PyTorch or JAX workflows. A practical starting point is to isolate one measurable particle, geometry or optimization operation and compare correctness and performance.

প্রধান বৈশিষ্ট্যসমূহ

  • Just-in-time compilation of Python kernels.
  • Supports CPU and CUDA execution devices.
  • Provides vector, matrix and geometry primitives.
  • Supports differentiable kernels and machine-learning integration.
  • Includes physics, geometry, optimization and finite-element examples.
  • Includes an example browser and USD animation output in some examples.

প্রয়োজনীয়তা, ইনস্টলেশন ও দ্রুত শুরু

1. Use Python 3.10 or newer.
2. Install pip install warp-lang; the package name is warp-lang.
3. Install pip install "warp-lang[examples]" for example dependencies.
4. Run python -m warp.examples.browse to explore examples.
5. Follow the README to define @wp.kernel functions, allocate wp.array data and launch work with wp.launch.
6. Validate numerical results on a small input before scaling the workload and measuring performance.

ব্যবহার

For a particle update, store positions and velocities in arrays, assign a particle to each thread and launch the kernel repeatedly to advance state. Compare the result with a simple reference implementation before timing larger runs. For parameter fitting, explore automatic differentiation and optimization examples and verify which operations support gradients in the intended pipeline.

Implementation notes
Separate compilation and warm-up time from steady-state measurements. Host-device transfers also cost time and can dominate small workloads. Arbitrary Python code is not automatically a valid Warp kernel; follow the supported type and compilation rules.

মডেলের সামঞ্জস্যতা ও ব্যবহারের ক্ষেত্র

Published wheels cover Windows x86-64, Linux x86-64/AArch64 and Apple Silicon macOS. CUDA acceleration on Windows and Linux needs a compatible NVIDIA GPU and driver. macOS wheels support CPU execution, not Metal acceleration.

লাইসেন্স ও ঝুঁকি-সংক্রান্ত নোট

Warp uses Apache-2.0. Some build dependencies and bundled components have separate terms listed in the repository’s licenses directory.

mediapipe

google-ai-edge/mediapipe

★ 36.9KC++

cosmos

NVIDIA/cosmos

★ 11.2KJupyter Notebook