Skip to content
Project categories

Machine Learning · GitHub

Train, evaluate, and run machine learning models.

Useful GitHub projects

6,537 projects

pdmd

ZeamoxWang

94stars
pdmd preview

Official Training and Benchmarking Implementation of PDMD: Projected Distribution Matching Distillation for Video Diffusion Models

scirex

zenteiq-research

18stars
scirex preview

JAX/Flax neural operators for scientific machine learning, with benchmarks and pretrained weights.

270stars
realtime-semantic-segmentation-pytorch preview

PyTorch implementation of over 30 realtime semantic segmentations models, e.g. BiSeNetv1, BiSeNetv2, CGNet, ContextNet, DABNet, DDRNet, EDANet, ENet, ERFNet, ESPNet, ESPNetv2, FastSCNN, ICNet, LEDNet, LinkNet, PP-LiteSeg, SegNet, ShelfNet, STDC, SwiftNet, and support knowledge distillation, distributed training, Optuna etc.

Machine LearningPythonApache-2.0GitHub

DRL-for-Path-Planning

zhaohaojie1998

604stars

深度强化学习路径规划, SAC路径规划, Soft Actor-Critic算法, SAC-pytorch,激光雷达Lidar避障,激光雷达仿真,Adaptive-SAC

Turbo-Learning

zhaozhen2333

30stars
Turbo-Learning preview

Official implementation of “A Turbo-Inference Strategy for Object Detection and Instance Segmentation” (CVIU 2026). A training-free framework for iterative refinement between object detection and instance segmentation.

  • No retraining: all refinement happens during inference.
  • Plug-and-play: the shared MaskBoxRefiner module is configurable from the RoI head.
Machine LearningPythonApache-2.0GitHub

Uformer

ZhendongWang6

950stars
Uformer preview

[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".

314stars

2019百度的关系抽取比赛,使用Pytorch实现苏神的模型,F1在dev集可达到0.75,联合关系抽取,Joint Relation Extraction.

RNNSharp

zhongkaifu

285stars
RNNSharp preview

RNNSharp is a toolkit of deep recurrent neural network which is widely used for many different kinds of tasks, such as sequence labeling, sequence-to-sequence and so on. It's written by C# language and based on .NET framework 4.6 or above versions. RNNSharp supports many different types of networks, such as forward and bi-directional network, sequence-to-sequence network, and different types of layers, such as LSTM, Softmax, sampled Softmax and others.

  • Tokyo - NNP - S_LOCATION
  • New - NNP - B_LOCATION
210stars

NIPS — Machine-Learning-Based Network Intrusion Prevention System | Kitsune/LUCID integrated | Online unsupervised anomaly detection | Kernel-level real-time traffic interception | FastAPI + CLI management

SimAM

ZjjConan

505stars

The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Networks".

PromptKG

zjunlp

735stars

PromptKG Family: a Gallery of Prompt Learning & KG-related research works, toolkits, and paper-list.

zyx

zk4x

74stars

Tensor library for machine learning

  • Eager mode, Lazy JIT Execution — tensor operations fuse into kernels as you write them; when fusion is no longer possible, the kernel executes. For one-off computations.
  • Tape (e-graph) — wrap loop bodies in a Tape for lazy graph building, autograd and e-graph-based fusion optimization. Computation happens when realize is called. For repeated computations.