Skip to content
Project categories

Machine Learning · GitHub

Train, evaluate, and run machine learning models.

Useful GitHub projects

7,238 projects

GiGL

Snapchat

86stars
GiGL preview

Gigantic Graph Learning (GiGL) Framework: Large-scale training and inference for Graph Neural Networks

  • 🧠 Versatile GNN Applications: Supports easy customization in using GNNs in supervised and unsupervised ML applications like node classification and link prediction.
  • 🚀 Designed for Scalability: The architecture is built with horizontal scaling in mind, ensuring cost-effective performance throughout the process of data preprocessing and transformation, model training, and inference.
737stars

推荐系统入门指南,全面介绍了工业级推荐系统的理论知识(王树森推荐系统公开课-基于小红书的场景讲解工业界真实的推荐系统),如何基于TensorFlow2训练模型,如何实现高性能、高并发、高可用的Golang推理微服务。Comprehensively introduced the theory of industrial recommender system, how to trainning models based on TensorFlow2, how to implement the high-performance、high-concurrency and high-available inference services base on Golang.

Machine LearningJupyter NotebookApache-2.0GitHub

treebank

someben

40stars

Ruby loader for the Stanford Sentiment Treebank (SST) parse trees and phrase labels, plus a locality-sensitive-hash index for word2vec binaries

VLM-AutoYOLO

Somnusochi

278stars
VLM-AutoYOLO preview

AI Auto Annotation & YOLO Training Pipeline, End-to-end object detection auto-labeling and YOLO training platform. VLM-powered annotation with NVIDIA LocateAnything-3B, manual refinement, one-click YOLO training, video keyframe extraction, and model validation. Supports image and video.

  • 🤖 VLM auto-labeling: Open-vocabulary object detection with LocateAnything-3B
  • 🎯 SAM2 / SAM3 segmentation: Bbox → pixel-precise mask with SAM 2.1 or SAM3 text-driven detection+segmentation in one pass, BBox/Mask toggle on canvas

tape

songlab-cal

746stars

Tasks Assessing Protein Embeddings (TAPE), a set of five biologically relevant semi-supervised learning tasks spread across different domains of protein biology.

Machine LearningPythonBSD-3-ClauseGitHub

kospeech

sooftware

637stars

Open-Source Toolkit for End-to-End Korean Automatic Speech Recognition leveraging PyTorch and Hydra.

Machine LearningPythonApache-2.0GitHub

iris-3b

speridlabs

227stars
iris-3b preview

Iris-3B: pixel-space generation & general vision learner. A 3B diffusion transformer that generates every pixel directly (no VAE); the same prior, fine-tuned, does monocular depth estimation and 4x image restoration. Paper, weights, training code and demo.

Machine LearningPythonApache-2.0GitHub

SimCLR

Spijkervet

824stars

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.