Попробовал taichi. Это такая штука, чтобы из питона рендерить на gpu. Строго говоря, это отдельный язык, которые встраивается в питон. Вот свою старую штуку переписал.
https://github.com/taichi-dev/taichi

#gif #taichi #processing