Nvidia Warp - Framework de Python para simulación y gráficos de alto rendimiento en GPU

• Framework de Python para escribir código de simulación y gráficos de alto rendimiento
• Compila funciones de Python con JIT y las convierte en código de kernel eficiente que puede ejecutarse en CPU o GPU
• Diseñado para computación espacial, ofrece varios primitivos para simulación física, robótica, procesamiento geométrico y más
• Los kernels de Warp son diferenciables, por lo que pueden usarse como parte de pipelines de aprendizaje automático como PyTorch y JAX
• Requisitos
  • Se recomienda Python 3.9 o superior
  • Soporta CPU x86-64 y ARMv8 en Windows, Linux y macOS
  • Para soporte de GPU, se requiere una GPU NVIDIA compatible con CUDA y sus drivers (mínimo GeForce GTX 9xx)
  • Se puede instalar desde PyPI
  • Los binarios compilados con runtime de CUDA 12.5 están disponibles en la página de GitHub Releases