PyTorch vs scikit-learn vs TensorFlow

PyTorch
PyTorch

187
115
9
scikit-learn
scikit-learn

349
166
24
TensorFlow
TensorFlow

960
396
58
- No public GitHub repository available -

What is PyTorch?

PyTorch is not a Python binding into a monolothic C++ framework. It is built to be deeply integrated into Python. You can use it naturally like you would use numpy / scipy / scikit-learn etc.

What is scikit-learn?

scikit-learn is a Python module for machine learning built on top of SciPy and distributed under the 3-Clause BSD license.

What is TensorFlow?

TensorFlow is an open source software library for numerical computation using data flow graphs. Nodes in the graph represent mathematical operations, while the graph edges represent the multidimensional data arrays (tensors) communicated between them. The flexible architecture allows you to deploy computation to one or more CPUs or GPUs in a desktop, server, or mobile device with a single API.

Want advice about which of these to choose?Ask the StackShare community!

Why do developers choose PyTorch?
Why do developers choose scikit-learn?
Why do developers choose TensorFlow?
What are the cons of using PyTorch?
What are the cons of using scikit-learn?
What are the cons of using TensorFlow?
    Be the first to leave a con
    What companies use PyTorch?
    What companies use scikit-learn?
    What companies use TensorFlow?
    What are some alternatives to PyTorch, scikit-learn, and TensorFlow?
    Keras
    Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on TensorFlow or Theano. https://keras.io/
    Caffe2
    Caffe2 is deployed at Facebook to help developers and researchers train large machine learning models and deliver AI-powered experiences in our mobile apps. Now, developers will have access to many of the same tools, allowing them to run large-scale distributed training scenarios and build machine learning applications for mobile.
    ML Kit
    ML Kit brings Google’s machine learning expertise to mobile developers in a powerful and easy-to-use package.
    TensorFlow.js
    Use flexible and intuitive APIs to build and train models from scratch using the low-level JavaScript linear algebra library or the high-level layers API
    PredictionIO
    PredictionIO is an open source machine learning server for software developers to create predictive features, such as personalization, recommendation and content discovery.
    See all alternatives
    What tools integrate with PyTorch?
    What tools integrate with scikit-learn?
    What tools integrate with TensorFlow?
      No integrations found
        No integrations found
        Decisions about PyTorch, scikit-learn, and TensorFlow
        No stack decisions found
        Interest over time
        Reviews of PyTorch, scikit-learn, and TensorFlow
        No reviews found
        How developers use PyTorch, scikit-learn, and TensorFlow
        Avatar of Yonas B.
        Yonas B. uses PyTorchPyTorch

        I used PyTorch when i was working on an AI application, image classification using deep learning.

        Avatar of Eliana Abraham
        Eliana Abraham uses TensorFlowTensorFlow

        Machine Learning in EECS 445

        Avatar of Eliana Abraham
        Eliana Abraham uses scikit-learnscikit-learn

        Machine Learning in EECS 445

        Avatar of Taylor Host
        Taylor Host uses TensorFlowTensorFlow

        Pilot integration for retraining.

        How much does PyTorch cost?
        How much does scikit-learn cost?
        How much does TensorFlow cost?
        Pricing unavailable
        Pricing unavailable
        Pricing unavailable
        News about PyTorch
        More news