I used conda / pip install on version 0.2.0_4. I faced the same issue. Code line -  “from . import lr_scheduler” was missing in the __ init __.py in the optim folder. I added it and after that I was able to import it.