FedSTM [TCCN] PyTorch implementation for Federated Learning with Stage-wise Trajectory Matching for UAV Networks on Non-IID Data (Early Access Article)