/distributed-mean-estimation

Primary LanguageJupyter NotebookMIT LicenseMIT

distributed-mean-estimation

PyPI Changelog License

Algorithms for Distributed Mean Estimation (DME) in PyTorch.

Installation

Install this library using pip:

pip install distributed-mean-estimation

Usage

Usage instructions go here.

Development

To contribute to this library, first checkout the code. Then create a new virtual environment:

cd distributed-mean-estimation
python -m venv venv
source venv/bin/activate

Now install the dependencies and test dependencies:

pip install -e ".[test]"

To run the tests:

pytest