I tried to install airio using the current setup.py, but the installation fails because the dependency grain==0.2.0 cannot be found or installed, therefore dependency airio does not install too.
The repository URL for grain in the dependency list seems outdated or missing (the old URL https://github.com/google-research/grain returns 404).
The current repository for grain appears to be here: https://github.com/google/grain
This issue is complicating the installation process on Windows.
Thanks in advance!