optimized nn.Sort. updates BlockMixture unit test