ENH: Watanabe-Akaike information criterion (WAIC) calculation