Software

Packages

Mauve (Documentation)
A package to compute the Mauve score for neural text generation. Install as pip install mauve-text. It is also supported via the HuggingFace Evaluate package.

SQwash (Documentation)
Distributionally robust learning in PyTorch with 1 additional line of code. Install as pip install sqwash.

Geom-Median
Fast and Differentiable Geometric Median in PyTorch and NumPy. Install as pip install geom-median.

Casimir (Documentation)
A toolbox of selected optimization algorithms for unstructured tasks such as binary classification, and structured prediction tasks such as visual object localization and named entity recognition.

RFA TensorFlow Federated Implementation of robust aggregation for federated learning using the geometric median.

Code to reproduce results from papers

Federated Learning with Partial Personalization: PyTorch implementation of various personalized federated learning algorithms and experiments on text, vision, and speech data. Reproduce results from this ICML 2022 paper.

Mauve Experiments: Implementation of Mauve and other similarity measures for neural text generation. Reproduce results from this NeurIPS 2021 paper.

RFA and PyTorch port tRFA: Implementation of RFA, a robust aggregation algorithm for federated learning, in simulation. Reproduce results from this paper published in IEEE Transactions on Signal Processing 2022.

Simplicial-FL: Implementation of Simplicial-FL to handle device heterogeneity in federated learning, in simulation. Reproduce results from this paper.