Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Core deep-learning frameworks and libraries for pretraining and distributed training.
Signals: deep-learning, neural-network, pytorch, tensorflow, jax, distributed-training, training, deepspeed
2,752 results
Deepfakes Software For All
AxisRL is an agentic RL post-training framework built on SGLang rollout, Megatron training, and real-world agent workflows.
The ultimate training toolkit for finetuning diffusion models
Tensors and Dynamic neural networks in Python with strong GPU acceleration
A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.
FlagGems is an operator library for large language models implemented in the Triton Language.
Visualizer for neural network, deep learning and machine learning models
An Open Source Machine Learning Framework for Everyone
GoMLX: An Accelerated Machine Learning Framework For Go
ncnn is a high-performance neural network inference framework optimized for the mobile platform
🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools
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
Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.
Graph Neural Network Library for PyTorch
High-performance automatic differentiation of LLVM and MLIR.
TabICLv2: A state-of-the-art tabular foundation model
Comprehensive open-source library of AI research and engineering skills for any AI model. Package the skills and your claude code/codex/gemini agent will be an AI research agent with full horsepower. Maintained by Orchestra Research.
Transform geospatial relations into graphs for Graph Neural Networks and spatial network analysis
The official CLI and Python client for the Hugging Face Hub.
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Supervisely SDK for Python - convenient way to automate, customize and extend Supervisely Platform for your computer vision task
Pretrained model hub for Keras 3.
A Python library for efficient Bayesian modeling with deep learning
Neural Network primitives with multiple backends
24,520 repositories in the index in total.
