Nvidia's TrackEi: Enhancing Rail Safety with AI