Last Layer Marginal Likelihood for Invariance Learning