TODO

[jon-chuang]

• Subgroup checks should only use batch if N > threshold. (e.g. not for single elements say PoK elem)
• impl specific details like batching for CPU and GPU should take place in batch_exp, not apply powers
• make uses of apply_powers completely sequential.