Official implementation for the paper: ...