Skip to content

Entry

PyTorch Geometric

Appears in 7 awesome lists

Graph neural network library for PyTorch enabling molecular modeling, materials discovery, protein interaction networks, and scientific knowledge graph learning (23.7k+ stars)

Open github.compyg-team/pytorch_geometric

Found in these lists

Awesome Ai For Science

Section: Specialized Frameworks · Graph neural network library for PyTorch enabling molecular modeling, materials discovery, protein interaction networks, and scientific knowledge graph learning (23.7k+ stars)

FreshScore 86

Awesome Data Analysis

Section: Tools · Geometric deep learning extension library for PyTorch.

FreshScore 80

AWESOME DATA SCIENCE

Section: Deep Learning Packages

FreshScore 92

Awesome Machine Learning

Section: Python · > Graph Neural Network Library for PyTorch.

FreshScore 93

Awesome Open Source AI

Section: 1. Core Frameworks & Libraries · Library for deep learning on irregular input data such as graphs, point clouds, and manifolds. Part of the PyTorch ecosystem.

FreshScore 89

Awesome Production Machine Learning

Section: Computation and Communication Optimisation · PyG (PyTorch Geometric) is a library built upon PyTorch to easily write and train Graph Neural Networks (GNNs) for a wide range of applications related to structured data.

FreshScore 92

awesome-python

Section: Other · Graph Neural Network Library for PyTorch

FreshScore 81

TensorFlow

How to use the Hexagon Delegate to speed up model inference on mobile and edge devices. Also see blog post Accelerating TensorFlow Lite on Qualcomm Hexagon DSPs.

In 23 listsDetails

PyTorch

(label: good first issue) PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing.

In 16 listsDetails

Netron

Visualizer for deep learning and machine learning models (no Python code, but visualizes models from most Python Deep Learning frameworks).

In 15 listsDetails

transformers

(formerly known as pytorch-transformers and pytorch-pretrained-bert) provides state-of-the-art general-purpose architectures (BERT, GPT-2, RoBERTa, XLM, DistilBert, XLNet, CTRL...) for Natural Language Understanding (NLU) and Natural Language Generation (NLG) with over 32+ pretrained models in…

In 14 listsDetails

XGBoost

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Flink and DataFlow. [Apache2]

In 11 listsDetails

TensorFlow-Slim

Official TensorFlow repository of state-of-the-art (SOTA) models and modeling solutions. Contains reference implementations for BERT, ResNet, Transformer, and many more with pre-trained weights and training scripts. Apache 2.0 licensed.

In 12 listsDetails

PyMC

Modern, comprehensive probabilistic programming framework in Python. Bayesian modeling with advanced MCMC sampling, variational inference, and seamless integration with ArviZ for visualization. Apache 2.0 licensed.

In 10 listsDetails

scikit-learn

Scikit-learn is a powerful machine learning library that provides a wide variety of modules for data access, data preparation and statistical model building.

In 10 listsDetails