winglian's picture
revert previous change and build ax images w docker on gpu (#371)
918f1b0 unverified