Duplicated from akhaliq/lama
b887a92
1
2
from .data_parallel import UserScatteredDataParallel, user_scattered_collate, async_copy_to
