Optimizing stimulus energy for cochlear implants with a machine learning model of the auditory nerve