ezpz.tp.mappings

• See ezpz/tp/mappings.py

ezpz/tp/mappings.py

Modified from: https://github.com/facebookresearch/fairscale/blob/5f484b3545f27eddb19d970fbe1d361b9c5f2b07/fairscale/nn/model_parallel/mappings.py