Automatic feature selection and weighting in molecular systems using Differentiable Information Imbalance