Skip to content
Project categories

Machine Learning · GitHub

Train, evaluate, and run machine learning models.

Useful GitHub projects

5,852 projects

torchcurves

alexshtf

74stars+1in 24h
torchcurves preview

Parametric differentiable curves with PyTorch for continuous embeddings, shape-restricted models, or KANs

  • Differentiable: Custom autograd function ensures gradients flow properly through the curve evaluation.
  • Vectorized: Vectorized operations for efficient batch and multi-curve evaluation.
Machine LearningPythonApache-2.0GitHub

PulmoScope

ali-119

7stars+1in 24h
PulmoScope preview

An end-to-end AI pipeline that screens lung CT slices for nodules and localizes them, using patient risk factors, a CNN classifier, and U-Net segmentation.

  • Multi-model decision pipeline. Three models with different roles are combined through explicit, auditable rules rather than a single opaque score.
  • Disagreement-aware output. When the classifier and segmenter disagree, the system returns uncertain and states which model flagged the slice.
Machine LearningJupyter NotebookMITGitHub

Axiom

aliasgharebrahimi

88stars+1in 24h
Axiom preview

Learn all the mathematics required for deep learning complete with examples and explanations of their specific applications in the field and implement every concept using PyTorch.

  • Mathematical Foundations: Detailed conceptual definitions paired with step-by-step solved examples.
  • PyTorch Implementation: All mathematical concepts are implemented using the PyTorch framework for practical learning.
Machine LearningPythonApache-2.0GitHub

MNISTVision

aliasgharebrahimi

95stars+1in 24h
MNISTVision preview

An optimized handwritten digit recognition model delivering exceptional accuracy, high-speed performance, and compatibility with low-end hardware, supported by comprehensive documentation, experiment tracking, data-driven decision-making, effective troubleshooting, and a modular project structure.

  • Maximum execution speed
  • Highly robust optimization
Machine LearningPythonApache-2.0GitHub

RePaint

andreas128

2,278stars+1in 24h
RePaint preview

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022

OpenNTN

ant-uni-bremen

97stars+1in 24h

3GPP TR 38.811 non-terrestrial network channel models for Sionna

  • Scenarios: dense urban (DenseUrban), urban (Urban) and suburban (SubUrban), with the parameter tables of TR 38.811 clause 6.7.2 and the fast-fading process of TR 38.901 clause 7.5.
  • Platforms: one satellite, from 160 km to 35 786 km, or one aerial platform, from 8 km to 50 km, per call, at the same height in every batch example.
Machine LearningJupyter NotebookMITGitHub

openfold

aqlaboratory

3,455stars+1in 24h
openfold preview

Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2

Machine LearningPythonApache-2.0GitHub

voxlock

arbazsiddiqui

111stars+1in 24h
voxlock preview

The best performing voice isolator there is. Keeps the caller's voice and removes the other voices in the room before speech-to-text hears them. Open source, streaming, runs on CPU and in the browser. pip install voxlock

Machine LearningPythonApache-2.0GitHub

einops

arogozhnikov

9,635stars+1in 24h

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Applying_EANNs

ArztSamuel

1,578stars+1in 24h
Applying_EANNs preview

A 2D Unity simulation in which cars learn to navigate themselves through different courses. The cars are steered by a feedforward neural network. The weights of the network are trained using a modified genetic algorithm.

146stars+1in 24h

🤖 Intelligent AI-powered PCB design automation tool using machine learning for component placement, routing optimization, and signal integrity analysis. Supports KiCad, Altium Designer, and Eagle CAD integration.

astroML

astroML

1,209stars+1in 24h

Machine learning, statistics, and data mining for astronomy and astrophysics

Machine LearningPythonBSD-2-ClauseGitHub

aTrain

aTrainTranscription

1,242stars+1in 24h
aTrain preview

A GUI tool for offline transcription of speech recordings, including speaker diarization, utilizing state-of-the-art machine learning models.

  • Batch Processing, allowing to have files queued for transcription
  • Add options for more verbatim output

OpenTranscribe

attevon-llc

113stars+1in 24h
OpenTranscribe preview

Self-hosted AI-powered transcription platform with speaker diarization, search, and collaboration features. Built with Svelte, FastAPI, and Docker for easy deployment.

  • Time-Stamped Comments: Add annotations at specific moments
  • User Management: Role-based access control (user / admin / super admin) with personalized settings