SplitFed An unofficial jax implementation of SplitFed: When Federated Learning Meets Split Learning (Thapa et al. 2021)