libtorch uses the parallel::data_parallel API as seen e.g. here 3. I don’t know if there are other wrappers, but based on the examples I would guess they should work in a similar way as the DataParallel wrapper in Python.