Machine Learning · GitHub

Train, evaluate, and run machine learning models.

GitHot

Useful GitHub projects

7,601 projectsUpdated --

6169

torch-sla

sparsexlab

59stars

torch sparse linear algebra library

Differentiable: gradients flow through solves, factorizations, and eigensolves via torch.autograd
Six verified backends: pytorch native Krylov (CPU/CUDA/ROCm), scipy (CPU), cudss (NVIDIA CUDA), strumpack direct (CPU/CUDA/ROCm), pyamg (CPU), amgx (NVIDIA CUDA) — each checked to ‖Ax−b‖/‖b‖ at or near machine precision

Machine LearningPythonApache-2.0GitHub

6170

spd_learn

spdlearn

42stars

SPDlearn: A Geometric Deep Learning Python Library for Neural Decoding Through Trivialization

SPD Layers — Specialized neural network layers for SPD matrices: BiMap for bilinear mappings, ReEig for eigenvalue rectification, and LogEig for tangent space projection.
Riemannian Operations — Complete toolkit for SPD manifold computations: exponential/logarithmic maps, geodesic distances, Log-Euclidean mean, and geodesic interpolation.

Machine LearningPythonBSD-3-ClauseGitHub

6171

finetuning-scheduler

speediedan

68stars

A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.

Machine LearningPythonApache-2.0GitHub

6172

CCNet

speedinghzl

1,486stars

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).

Machine LearningPythonMITGitHub

6173

pytorch-segmentation-toolbox

speedinghzl

776stars

PyTorch Implementations for DeeplabV3 and PSPNet

Fewness of Training Time
Better Reproduced Performance

Machine LearningPythonMITGitHub

6174

CUDA-for-AMD-Windows

Speedstu

334stars

Run CUDA-targeted Windows applications on AMD GPUs with ZLUDA + ROCm/HIP.

Machine LearningPythonGitHub

6175

armiger

spencernorton

97stars

Local ONNX classification of torrent and media metadata, with optional catalogue and swarm estimates.

Machine LearningPythonApache-2.0GitHub

6176

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

6177

SimCLR

Spijkervet

824stars

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

Machine LearningPythonMITGitHub

6178

SNN-Daily-Arxiv

SpikingChen

524stars

Update arXiv papers about Spiking Neural Networks daily.

Machine LearningPythonGitHub

6179

pedalboard

spotify

6,482stars

🎛 🔊 A Python library for audio.

Support for reading and writing AIFF, FLAC, MP3, OGG, and WAV files on all platforms with no dependencies
Additional support for reading AAC, AC3, WMA, and other formats depending on platform

Machine LearningC++GPL-3.0GitHub

6180

voyager

spotify

1,608stars

🛰️ An approximate nearest-neighbor search library for Python and Java with a focus on ease of use, simplicity, and deployability.

Machine LearningC++Apache-2.0GitHub

6181

elasticdl

sql-machine-learning

743stars

Kubernetes-native Deep Learning Framework

Machine LearningPythonMITGitHub

6182

sqlflow

sql-machine-learning

5,187stars

Brings SQL and AI together.

Machine LearningGoApache-2.0GitHub

6183

datasets

src-d

347stars

source{d} datasets ("big code") for source code analysis and machine learning on source code

Structural commit features
Description: 1.6 million commits in 622 Java repositories on GitHub.

Machine LearningJupyter NotebookGitHub

6184

hercules

src-d

2,812stars

Gaining advanced insights from Git repository history.

Machine LearningGoGitHub

6185

kmcuda

src-d

841stars

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA

Machine LearningJupyter NotebookGitHub

6186

PFNN

sreyafrancis

525stars

Phase-Functioned Neural Networks for Character Control

Machine LearningPythonGitHub

6187

machine-learning-interview-questions

Sroy20

1,671stars

This repository is to prepare for Machine Learning interviews.

Machine Learning GitHub

6188

annotated-s4

srush

520stars

Implementation of https://srush.github.io/annotated-s4

Machine LearningPythonMITGitHub

6189

Tensor-Puzzles

srush

4,451stars

Solve puzzles. Improve your pytorch.

Machine LearningJupyter NotebookMITGitHub

6190

eesen

srvk

836stars

The official repository of the Eesen project

The WFST-based decoding approach can incorporate lexicons and language models into CTC decoding in an effective and efficient way.
The RNN-LM decoding approach does not require a fixed lexicon.

Machine LearningC++Apache-2.0GitHub

6191