neural-estimator

InfoAtlas is a foundation model that directly estimates mutual information in a single forward pass, achieving 100x speedup over traditional neural estimators while matching accuracy. It is pretrained on synthetic data and generalizes to real-world scenarios.