Deep Learning Framework

  • Pathways: Asynchronous Distributed Dataflow for ML (MLSys 2022) [Paper]

    • Google

    • Outstanding Paper Award

  • OneFlow: Redesign the Distributed Deep Learning Framework from Scratch (arXiv 2110.15032) [Paper] [Code] [中文官网]

    • OneFlow

  • Jittor: a novel deep learning framework with meta-operators and unified graph execution (Science China Information Sciences 2020) [Paper] [Code] [中文主页]

    • THU

  • PyTorch: An Imperative Style, High-Performance Deep Learning Library (NeurIPS 2019) [Paper] [Code] [Homepage]

    • FAIR

  • XDL: An Industrial Deep Learning Framework for High-dimensional Sparse Data (DLP-KDD 2019) [Paper] [Code]

    • Alibaba

    • High-dimensional sparse data.

  • TensorFlow: A System for Large-Scale Machine Learning (OSDI 2016) [Paper] [Code] [Homepage]

    • Google Brain

  • MXNet: A Flexible and Efficient Machine Learning Library for Heterogeneous Distributed Systems (NIPS 2016 Workshop on MLSys) [Paper] [Homepage] [Code]

  • Caffe: Convolutional Architecture for Fast Feature Embedding (arXiv 1408.5093) [Paper] [Homepage] [Code]

    • UC Berkeley

  • An Introduction to Computational Networks and the Computational Network Toolkit (MSR-TR-2014-112) [Paper] [Code] [Homepage]

    • Microsoft

    • CNTK

    • No longer developed.

Last updated