While I never made my research code public, https://openreview.net/forum?id=SyxytxBFDr documents a similar setup to what I use.