Enhancing Optimization Algorithms For Non-Convex Settings : Federated And Decentralized Learning