Open-source alternatives to Graph Nets

This repository serves as an educational resource for implementing graph neural networks using Python. It provides a collection of structured code examples and tutorials designed to guide developers through the process of building and training machine learning models that operate on complex, interconnected datasets. The project covers the core mechanics of graph-based deep learning, including message-passing architectures, feature aggregation, and the stacking of convolutional layers. It demonstrates how to represent non-Euclidean data as static graphs and how to manage memory during training

DGL is a Python library for building and training graph neural networks. It functions as a graph message passing framework and a geometric deep learning tool, enabling the development of models that analyze graph-structured data. The library is designed for large-scale graph processing, utilizing distributed training and neighbor sampling to handle datasets with billions of edges. It provides specialized support for heterogeneous graph modeling, allowing for the representation of complex real-world entities with multiple node and edge types. Its capabilities cover a wide range of graph tasks

pygcn is a PyTorch library and framework for implementing graph convolutional networks. It provides tools for semi-supervised node classification and the generation of node embeddings from graph-structured data. The system converts graph nodes into low-dimensional vectors based on neighborhood patterns and local similarities. It enables the prediction of node labels by leveraging both a small set of labeled examples and the overall graph topology. The library covers relational data analysis and semi-supervised graph learning. It includes computational primitives for message passing, adjacenc

Cnn_graph is a graph convolutional network framework and graph signal processing library designed for machine learning research. It provides computational notebooks and code to process and classify graph-structured data by combining node features with an underlying adjacency matrix representation. The framework performs spectral graph convolutions through localized filters and accelerates filtering operations using truncated Chebyshev polynomials to avoid explicit graph Laplacian diagonalization. It includes a graph-structured data pipeline and sparse adjacency representations to handle irreg

Memgraph is an in-memory, distributed graph database designed for high-performance labeled property graph management. It utilizes a Cypher query engine for declarative data retrieval and manipulation, providing a scalable knowledge graph backend that integrates vector search and graph traversals. The system distinguishes itself as a real-time graph analytics platform, employing native C++ and CUDA implementations to execute complex network analysis and dynamic community detection on streaming data. It provides specialized support for AI integration, including GraphRAG capabilities, the constr

This project is a collection of neural network models and geometric tools designed for image feature matching, spatial alignment, and visual localization. It provides a pre-trained neural network model for identifying high-accuracy correspondences between sparse image features without requiring local training. The system utilizes a graph neural network matcher that employs attention mechanisms and message passing to learn spatial relationships between image feature points. It integrates a RANSAC camera pose estimator to filter feature matches and calculate the relative spatial transformation

GraphTransformerNetworks is a graph neural network framework implemented in PyTorch for learning structural representations and performing classification tasks on complex heterogeneous graphs and relational networks. The project provides automated preprocessing pipelines to transform raw graph datasets into standardized formats, alongside model training, forward passes, and gradient backpropagation executed through dynamic tensor operations. The architecture incorporates self-attention mechanisms applied directly to graph structures to learn contextual representations of nodes and edges ac

OpenKE is a knowledge graph embedding framework designed to transform structured knowledge graphs into low-dimensional vector representations. It functions as a library for representation learning and a toolset for converting entities and relations into numerical embeddings. The project includes a link prediction engine to evaluate the likelihood of relationships between entities and identify missing facts in large-scale graphs. It provides a dedicated preprocessing tool to map raw entity and relation strings into numerical identifiers for machine learning training. The framework's capabilit

This project is a PyTorch person re-identification framework designed for training and evaluating models that identify individuals across different camera views. It provides a complete model training pipeline, a deep learning feature extractor for converting images into numeric vectors, and a suite of computer vision benchmarking tools to measure identity retrieval accuracy. The framework includes a specialized transfer learning toolkit that supports layer freezing, staged learning rate optimization, and differential learning rates for fine-tuning pretrained models. It distinguishes itself th

KnowledgeGraphCourse is a structured collection of graduate-level academic materials, lecture notes, and a comprehensive curriculum focused on the theory and application of knowledge graphs. It serves as a markdown-based educational resource that provides navigable course modules and study guides. The material covers specialized research on integrating knowledge graphs with large language models to reduce hallucinations. It includes detailed guides on using the SPARQL language for storing large-scale graph datasets and executing optimized queries. The curriculum spans a broad range of capabi

AllenNLP is a PyTorch-based research library and deep learning language toolkit designed for developing and training neural network architectures for linguistic tasks. It provides a distributed training system that coordinates data and gradients across multiple GPUs and a framework for integrating pretrained transformer architectures. The system distinguishes itself with a dedicated algorithmic bias mitigation tool used to identify and reduce bias in linguistic model predictions. It also includes model influence analysis to interpret predictions by calculating the influence of specific traini

This project is a PyTorch implementation of the EfficientDet architecture designed for real-time object detection. It provides a neural network and inference engine capable of identifying and locating multiple objects within images or video streams. The implementation includes pretrained computer vision models with optimized weights, enabling immediate inference and fine-tuning without the need for training from scratch. The project covers the full pipeline for computer vision model optimization, including custom object detection training and model weight optimization. It incorporates struct

CV-Backbones is a computer vision backbone library and model zoo providing a collection of pre-defined neural network architectures for extracting visual features and processing image data. It serves as a PyTorch vision framework of reusable deep learning components designed for image analysis and visual representation learning. The library focuses on efficient neural network architectures to reduce computational overhead while maintaining feature extraction performance. This is achieved through the implementation of lightweight model designs such as GhostNet and MLP. The project covers a br

BigDL is a PyTorch acceleration framework and distributed inference engine designed for large language models. It provides a toolkit for running models on Intel hardware, integrating quantization tools and libraries for parameter-efficient fine-tuning. The project distinguishes itself through the use of pipeline parallelism to distribute model workloads across multiple hardware accelerators. It utilizes low-bit integer quantization and speculative decoding to reduce memory footprints and decrease text generation latency. The system covers broad capabilities in model optimization, including w

This library provides a comprehensive collection of modular building blocks and research-backed architectures for implementing vision transformers within the PyTorch framework. It serves as a centralized repository for constructing, training, and analyzing attention-based models, offering a wide array of specialized variants designed for image classification and visual representation learning. The project distinguishes itself through a focus on architectural efficiency and flexibility, supporting diverse input formats including non-square images and volumetric data like video. It incorporates

DocTR is a deep learning OCR library built on PyTorch that detects and transcribes text in document images using a two-stage detection-recognition pipeline. It provides a complete framework for building and deploying OCR pipelines with pretrained models available through the Hugging Face Hub, and supports exporting trained models to ONNX format for cross-runtime deployment. The library offers end-to-end OCR pipelines that combine text detection and recognition to extract all text from document images or PDFs, with support for rotated page handling and varied text orientations. It includes cap

SimSwap is a deep learning face swapping framework and computer vision media processor built with PyTorch. It functions as an image synthesis tool designed to replace a person's identity in images and videos with a target face using a single trained model. The system operates as a video identity replacement tool that swaps identities across frames while preserving the original expressions and lighting of the source media. It enables digital identity manipulation and the production of synthetic media through automated facial feature mapping. The framework supports both the application of trai

Kaolin is a PyTorch 3D deep learning library providing a comprehensive suite of tools for 3D geometry processing, physics simulation, data visualization, and gradient-based rendering for computer vision. The library includes a differentiable 3D renderer and a geometry processing toolkit for converting and transforming 3D representations such as meshes and point clouds. It also features a 3D physics simulation engine to calculate physical interactions and collisions between three-dimensional objects and scenes. The toolkit provides utilities for 3D data visualization, including the creation o

This project is a deep learning curriculum and a collection of PyTorch tutorials designed for deep learning education. It provides a structured set of technical documents and runnable notebooks that translate theoretical machine learning concepts into executable code. The repository includes implementation guides for various neural network architectures, specifically covering convolutional, recurrent, and transformer-based models. It provides practical examples for building computer vision pipelines for object detection and semantic segmentation, as well as natural language processing tools f

AudioLCM is a deep learning framework designed for text-to-audio synthesis. It functions as a generative engine that converts written descriptions into high-fidelity audio clips by processing text prompts through latent consistency models. The project distinguishes itself by utilizing latent consistency distillation to enable rapid audio generation. By mapping diffusion trajectories to a single-step consistency function, the system achieves efficient sound synthesis while maintaining the output quality typically associated with iterative diffusion processes. The framework provides a comprehe

This PyTorch-based deep learning library provides a framework for analyzing and forecasting temporal data. It implements specialized architectures for time series forecasting, anomaly detection, data imputation, and classification. The project distinguishes itself through the inclusion of zero-shot inference capabilities, allowing large-scale temporal models to be evaluated on unseen datasets without requiring task-specific fine-tuning. The framework covers a broad range of analytical capabilities, including the recovery of missing values in incomplete datasets, the identification of irregul

This project is a collection of optimized scripts, deployment patterns, and reference implementations designed for scaling and accelerating state-of-the-art AI models. It serves as a multi-domain model zoo and a distributed training framework, providing PyTorch reference implementations for training and deploying models on GPU-accelerated infrastructure. The repository distinguishes itself through an optimization suite focused on NVIDIA GPU hardware, utilizing automatic mixed precision and specialized math modes to increase training speed and throughput. It provides enterprise deployment patt

This repository provides a comprehensive educational framework for mastering machine learning and deep learning through a structured curriculum. It integrates theoretical mathematical foundations—including calculus, probability, and linear algebra—with hands-on laboratory implementations that require learners to build algorithms and neural network architectures from scratch. The project distinguishes itself by emphasizing first-principles development, ensuring that students understand the underlying mechanics of backpropagation, layer-wise computation, and model optimization. It covers a broa

A curated list of network embedding techniques.

StellarGraph - Machine Learning on Graphs

Tutorial: Graph Neural Networks for Natural Language Processing at EMNLP 2019 and CODS-COMAD 2020

Must-read papers on knowledge representation learning (KRL) / knowledge embedding (KE)