Machine learning for autonomous driving