Nvidia Warp - Framework Python pour la simulation et les graphismes GPU haute performance

• Framework Python pour écrire du code de simulation et de graphisme haute performance
• Compile à la volée des fonctions Python avec le JIT afin de les transformer en code de noyau efficace pouvant s'exécuter sur CPU ou GPU
• Conçu pour l'informatique spatiale, avec diverses primitives pour la simulation physique, la robotique, le traitement géométrique, etc.
• Les noyaux Warp sont différentiables, ce qui permet de les utiliser dans des pipelines de machine learning comme PyTorch et JAX
• Exigences
  • Python 3.9 ou version ultérieure recommandé
  • Prise en charge des CPU x86-64 et ARMv8 sur Windows, Linux et macOS
  • Pour la prise en charge GPU, un GPU NVIDIA compatible CUDA et ses pilotes sont requis (minimum GeForce GTX 9xx)
  • Installation possible via PyPI
  • Des binaires compilés avec le runtime CUDA 12.5 sont fournis sur la page GitHub Releases