Discrete Key-Value Bottleneck