if __name__ == "__main__":
    import bagua  # noqa: F401
    import deepspeed  # noqa: F401