miniFederate is a minimal library that implements the federated averaging step for PyTorch models.
Please note that this is purely written for experimental fun. In order to actually use this in an application, you'll need to add networking functionality.