https://mathpn.com/posts/exploring-fractals-with-pytorch/