PyTorch: An Imperative Style, High-Performance Deep Learning Library.
Adam PaszkeSam GrossFrancisco MassaAdam LererJames BradburyGregory ChananTrevor KilleenZeming LinNatalia GimelsheinLuca AntigaAlban DesmaisonAndreas KöpfEdward Z. YangZachary DeVitoMartin RaisonAlykhan TejaniSasank ChilamkurthyBenoit SteinerLu FangJunjie BaiSoumith ChintalaPublished in: NeurIPS (2019)