Travis Head got Australia off to a flying start against Sri Lanka as the visitors pile on the pressure in the first Test.