Getting an all-optical AI to handle non-linear math