sgugger's picture
Handle model parallelism
0dc4dba