Vision processing unit for machine learning