RealTime Mandelbrot set with python pygame and taichi. code in the comments