ENH: fix infomax + deprecate + generalzie API