Improving computation efficiency using input and architecture features for a virtual screening application