Imported from https://github.com/belkakari/cellular-automata-pytorch