Non-linear contextual bandits