neural discrete representation learning github