Create krizhevsky_layers.py