Skip to content
Project categories

Machine Learning · GitHub

Train, evaluate, and run machine learning models.

Useful GitHub projects

6,913 projects

knockknock

huggingface

2,826stars+1in 24h

🚪✊Knock Knock: Get notified when your training ends with only two additional lines of code

optimum

huggingface

3,520stars+1in 24h

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy-to-use hardware optimization tools

Machine LearningPythonApache-2.0GitHub
37,284stars+1in 24h

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Machine LearningPythonApache-2.0GitHub

transformers.js

huggingface

16,417stars+1in 24h

State-of-the-art Machine Learning for the web. Run 🤗 Transformers directly in your browser, with no need for a server!

Machine LearningJavaScriptApache-2.0GitHub

gocv

hybridgroup

7,519stars+1in 24h
gocv preview

Go package for computer vision using OpenCV 4 and beyond. Includes support for DNN, CUDA, OpenCV Contrib, and OpenVINO.

530stars+1in 24h
aihwkit preview

IBM Analog Hardware Acceleration Kit

  • A library of device presets that are calibrated to real hardware data and based on models in the literature, along with a configuration that specifies a particular device and optimizer choice.
  • A module for executing high-level use cases ("experiments"), such as neural network training with minimal code overhead.
Machine LearningJupyter NotebookMITGitHub

DINO

IDEA-Research

2,865stars+1in 24h
DINO preview

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"

Machine LearningPythonApache-2.0GitHub

Cam2BEV

ika-rwth-aachen

794stars+1in 24h
Cam2BEV preview

TensorFlow Implementation for Computing a Semantically Segmented Bird's Eye View (BEV) Image Given the Images of Multiple Vehicle-Mounted Cameras.

108stars+1in 24h

PyTorch ML quant research: 213 factors, bias correction, portfolio optimization, backtesting, and System-One/Jev-style decisions.

2,731stars+1in 24h

SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime

Machine LearningPythonApache-2.0GitHub

mtcnn

ipazc

2,493stars+1in 24h
mtcnn preview

MTCNN face detection implementation for TensorFlow, as a PIP package.

Machine LearningJupyter NotebookMITGitHub

iree

iree-org

4,013stars+1in 24h
iree preview

A retargetable MLIR-based machine learning compiler and runtime toolkit.