Lenia-in-Pytorch pytorch based example of Lenia requirements pygame pytorch (>=2.0 recommended) numpy