Skip to content
Project categories

Machine Learning · GitHub

Train, evaluate, and run machine learning models.

Useful GitHub projects

4,297 projects

doctr

mindee

6,424stars
doctr preview

docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning. Ongoing development and maintenance by t2k.

Machine LearningPythonApache-2.0GitHub

code

ml-from-scratch-book

325stars
code preview

Companion code for Machine Learning From Scratch — 10 core ML algorithms built from scratch with NumPy, compared with Scikit-learn and PyTorch.

Machine LearningJupyter NotebookMITGitHub

river

online-ml

6,141stars
river preview

🌊 Online machine learning in Python

  • Linear models, with a wide array of optimizers
  • Decision trees and random forests
Machine LearningPythonBSD-3-ClauseGitHub

DIFFormer

qitianwu

312stars
DIFFormer preview

The official implementation for ICLR2023 spotlight paper "DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusion"

NodeFormer

qitianwu

311stars
NodeFormer preview

The official implementation of NeurIPS2022 spotlight paper "NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classification"

  • All-Pair Message Passing on Layer-specific Adaptive Structures. The feature propagation per layer is operated on a latent graph that potentially connect all the nodes, in contrast with the local propagation design of GNNs that only aggregates the embeddings of neighbored nodes.
  • Efficient End-to-End Learning for Latent Structures. The optimization for the latent structures is allowed for end-to-end training with the model, making the whole learning process simple and efficient. E.g., the training on Cora only requires 1-2 minutes, while on OGBN-Proteins requires 1-2 hours in one run.
Machine LearningJupyter NotebookGitHub

nanodet

RangiLyu

6,276stars
nanodet preview

NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥

Machine LearningPythonApache-2.0GitHub
28stars
inno-kokoro preview

Zero-shot clone-tuner for Kokoro-82M. Generate stock-compatible custom voice packs from 5-seconds of reference audio in less than 1-second

Machine LearningPythonApache-2.0GitHub
347stars
SciMLBenchmarks.jl preview

Scientific machine learning (SciML) benchmarks, AI for science, and (differential) equation solvers. Covers Julia, Python (PyTorch, Jax), MATLAB, R

skorch

skorch-dev

6,184stars

A scikit-learn compatible neural network library that wraps PyTorch

Machine LearningJupyter NotebookBSD-3-ClauseGitHub

Lihang

SmirkCao

6,331stars
Lihang preview

Statistical learning methods, 统计学习方法(第2版)[李航] [笔记, 代码, notebook, 参考文献, Errata, lihang]