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