Hi Sean, thank you for coming back to me and my apologies for my delayed response. Unfortunately, those state callbacks are necessary for Elastic Horovod to work (https://horovod.readthedocs.io/en/stable/elastic_include.html) which I want to use as it allows me to use spot VMs that greatly reduce cost.

In case I am using it wrong or if there is another thing I should do to make elastic Horovod and horovodrunner work together, let me know and I'll give it a try!