Eq. 19 in this paper gives the following Hamiltonian: $\sigma_a, \tau_a, \eta_a$ are respectively the spin, sublattice pseudospin and valley pseudospin respectively. Normally, I would have chosen a basis convention such as $\sigma_a, \tau_a, \eta_a$ and plugged in the $2\times 2$ identity matrix...