apex/transformer/tensor_parallel/cross_entropy.py · f79993d92cfefbaf20a670dcbd61749ac70aed28 · OpenDAS / apex

Masaki Kozuki authored Oct 02, 2021


Co-authored-by: Piotr Bialecki <pbialecki@nvidia.com>
Co-authored-by: Eddie Yan <eddiey@nvidia.com>
Co-authored-by: Rishi Puri <riship@nvidia.com>
Co-authored-by: Sangkug Lym <slym@nvidia.com>

365fdc18

cross_entropy.py 4.53 KB

Replace cross_entropy.py