Nvidia Warp - framework Python para simulação e gráficos de alto desempenho em GPU

• Framework Python para escrever código de simulação e gráficos de alto desempenho
• Compila funções Python com JIT e as converte em código de kernel eficiente, executável em CPU ou GPU
• Projetado para computação espacial, oferecendo diversos primitivos para simulação física, robótica, processamento geométrico e mais
• Os kernels do Warp são differentiable, podendo ser usados como parte de pipelines de machine learning como PyTorch e JAX
• Requisitos
  • Python 3.9 ou superior recomendado
  • Suporte a CPUs x86-64 e ARMv8 no Windows, Linux e macOS
  • Para suporte a GPU, é necessário uma GPU NVIDIA com suporte a CUDA e drivers compatíveis (mínimo GeForce GTX 9xx)
  • Disponível para instalação via PyPI
  • Binários compilados com o runtime CUDA 12.5 estão disponíveis na página de GitHub Releases