Nvidia Warp - 고성능 GPU 시뮬레이션 및 그래픽을 위한 Python 프레임워크
(github.com/NVIDIA)- 고성능 시뮬레이션 및 그래픽 코드를 작성하기 위한 Python 프레임워크
- Python 함수를 JIT 컴파일하여 CPU 또는 GPU에서 실행 가능한 효율적인 커널 코드로 변환
- 공간 컴퓨팅을 위해 설계되었으며, 물리 시뮬레이션, 로보틱스, 기하학 처리 등을 위한 다양한 프리미티브를 제공
- Warp 커널은 differentiable 하여 PyTorch 및 JAX와 같은 머신러닝 파이프라인의 일부로 사용 가능
- 요구사항
- Python 3.9 이상 권장
- Windows, Linux, macOS에서 x86-64 및 ARMv8 CPU 지원
- GPU 지원을 위해 CUDA 지원 NVIDIA GPU 및 드라이버 필요 (최소 GeForce GTX 9xx)
- PyPI에서 설치 가능
- CUDA 12.5 런타임으로 빌드된 바이너리는 GitHub Releases 페이지에서 제공