Light propagation can be controlled at the nanoscale by forming surface polaritons in two-dimensional layers of van der Waals crystals stacked with a relative twist. Now, a neural network that works backwards from a target light path identifies structures that produce a desired propagation across visible to terahertz operating frequencies.