NVIDIA Warp 기반의 GPU 가속 물리 시뮬레이션 엔진으로, 로봇공학자와 시뮬레이션 연구자를 위한 프로젝트입니다.
NVIDIA/warp
일반 Python 함수를 CPU 또는 NVIDIA GPU에서 실행되는 커널로 JIT 컴파일하는 Python 프레임워크로, 시뮬레이션, 로보틱스, ML에 활용할 수 있습니다.
- 스타
- 7,273
- 포크
- 682
- 열린 이슈
- 390
- 마지막 커밋
- 12시간 전
어떤 프로젝트인가요?
Warp는 일반 Python 함수를 효율적인 커널 코드로 JIT 컴파일해 CPU 또는 GPU에서 실행합니다. GPU 가속 시뮬레이션, 로보틱스, 기하 처리, 머신러닝을 대상으로 하며, 물리 및 기하 작업을 위한 프리미티브를 제공합니다.
Warp 커널은 미분 가능하므로 PyTorch, JAX, Paddle과 함께 머신러닝 파이프라인에서 사용할 수 있습니다. 패키지는 PyPI의 warp-lang wheel로 배포되며, Windows와 Linux x86-64에서는 CUDA 가속을 지원합니다. macOS용 wheel은 CPU에서만 실행됩니다.
저장소에는 물리 시뮬레이션, 유한 요소법, 미분 가능 최적화, 타일 기반 GPU 프로그래밍을 다루는 다양한 예제와 Colab에서 열 수 있는 튜토리얼 노트북이 포함되어 있습니다.
핵심 기능
- 01일반 Python 함수를 CPU 또는 CUDA GPU에서 실행되는 커널로 컴파일합니다.
- 02커널은 미분 가능하며 PyTorch, JAX, Paddle과 연동됩니다.
- 03물리 시뮬레이션, 기하 처리, 로보틱스를 위한 프리미티브를 제공합니다.
- 04PyPI에서 warp-lang으로 설치할 수 있으며, 예제용 선택적 의존성도 제공됩니다.
- 05시뮬레이션과 최적화를 다루는 예제 및 튜토리얼 노트북이 포함되어 있습니다.
활용 사례
- Python에서 NVIDIA GPU로 대규모 입자 또는 유체 시뮬레이션을 실행합니다.
- 미분 가능한 시뮬레이터를 구축해 PyTorch 또는 JAX 학습 루프에 그래디언트를 전달합니다.
- Python으로 메시 레이캐스팅이나 등가면 추출 같은 기하 처리를 수행합니다.
- CUDA C++를 직접 작성하지 않고 GPU 커널을 프로토타이핑합니다.
빠른 시작
- 1
PyPI에서 Warp를 설치합니다
pip install warp-lang - 2
Quick Start 코드의 중력 시뮬레이션 예제를 실행합니다(Python 파일로 저장한 다음 실행).
사용 전 알아둘 점
- Linux 및 Windows에서 CUDA 가속을 사용하려면 Turing(sm_75) 이상인 NVIDIA GPU와 최신 NVIDIA 드라이버가 필요합니다.
- macOS용 wheel은 CPU에서만 실행되며 Metal 가속을 지원하지 않습니다.
- CUDA 12 환경에서는 별도의 +cu12 wheel을 사용하거나 소스에서 빌드해야 합니다.
포함된 컬렉션
토픽
자주 묻는 질문
- warp은(는) 무엇인가요?
- 일반 Python 함수를 CPU 또는 NVIDIA GPU에서 실행되는 커널로 JIT 컴파일하는 Python 프레임워크로, 시뮬레이션, 로보틱스, ML에 활용할 수 있습니다. Warp는 일반 Python 함수를 효율적인 커널 코드로 JIT 컴파일해 CPU 또는 GPU에서 실행합니다. GPU 가속 시뮬레이션, 로보틱스, 기하 처리, 머신러닝을 대상으로 하며, 물리 및 기하 작업을 위한 프리미티브를 제공합니다.
- warp은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
- warp은(는) Apache-2.0 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
- warp은(는) 아직 유지보수되고 있나요?
- 활발히 유지보수 중. warp의 마지막 커밋은 2026년 11월 5일입니다.
- warp은(는) 어디에 사용할 수 있나요?
- Python에서 NVIDIA GPU로 대규모 입자 또는 유체 시뮬레이션을 실행합니다. 미분 가능한 시뮬레이터를 구축해 PyTorch 또는 JAX 학습 루프에 그래디언트를 전달합니다. Python으로 메시 레이캐스팅이나 등가면 추출 같은 기하 처리를 수행합니다. CUDA C++를 직접 작성하지 않고 GPU 커널을 프로토타이핑합니다.
- warp는 어떻게 설치하나요?
- PyPI에서 Warp를 설치합니다: pip install warp-lang Quick Start 코드의 중력 시뮬레이션 예제를 실행합니다(Python 파일로 저장한 다음 실행).
- warp은(는) 어디서 받을 수 있나요?
- 소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/NVIDIA/warp
요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.
관련 프로젝트
NVIDIA Warp 기반의 GPU 가속 MuJoCo 물리 시뮬레이터로, 로보틱스 시뮬레이션을 높은 처리량으로 실행합니다.
JAX 기반 강체 물리 엔진으로, 로보틱스 및 RL 연구를 위한 대규모 병렬 미분 가능한 시뮬레이션을 가속기에서 실행합니다.
Crazyflie 2.x 드론을 대상으로 CPU와 GPU에서 배치형 미분 가능 시뮬레이션을 실행하는 JAX 기반 쿼드로터 시뮬레이터입니다.
NVIDIA GPU에서 딥러닝 추론 성능을 높여 주는 SDK, TensorRT의 오픈 소스 구성 요소로 플러그인, ONNX 파서, 샘플을 포함합니다.
로봇, 차량, 인체 골격 같은 다관절 시스템을 시뮬레이션하는 오픈소스 C++ 다물체 동역학 라이브러리입니다.