Learning Cellular Automaton Dynamics with Neural Networks