viztracer can create traces for python programs. Not sure about async-io awareness though.