Fixing some core bugs

Updates to parallel processing, key similarity parameters, and more

Generalized DSA and InputDSA

Updates based on our new ICLR paper and upcoming work