kubeflow/trainer
quality grade A, 91 out of 100Distributed AI Model Training and LLM Fine-Tuning on Kubernetes
- stars
- 2.2k
- stars gained this week
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,672 results
Distributed AI Model Training and LLM Fine-Tuning on Kubernetes
A retargetable MLIR-based machine learning compiler and runtime toolkit.
A lightweight library for PyTorch training tools and utilities
Containers for machine learning
A machine learning compiler for GPUs, CPUs, and ML accelerators
GHOSTS (General Human-Oriented Synthetic Teammates and Systems) is a realistic user simulation framework for cyber experimentation, simulation, training, and exercise
Translate manga/image 一键翻译各类图片内文字 https://cotrans.touhou.ai/ (no longer working)
JAX-based neural network library
Pointcept: Perceive the world with sparse points, a codebase for point cloud perception research. Latest works: Utonia (ICML'26), Concerto (NeurIPS'25), Sonata (CVPR'25 Highlight), PTv3 (CVPR'24 Oral)
🔥 LeetCode for PyTorch — practice implementing softmax, attention, GPT-2 and more from scratch with instant auto-grading. Jupyter-based, self-hosted or try online.
Graph-based weather forecasting models. Originally, PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.07575)
A high-performance machine learning library in pure Rust, offering statistical utilities, ML algorithms and neural networks.
Machine learning algorithms for many-body quantum systems
A Julia machine learning framework
A toolkit for time series machine learning and deep learning
Kubernetes Operator for MPI-based applications (distributed training, HPC, etc.)
VisualTorch aims to help visualize Torch-based neural network architectures.
Graph Neural Networks in Julia
A component of the DiffEq ecosystem for enabling sensitivity analysis for scientific machine learning (SciML). Optimize-then-discretize, discretize-then-optimize, adjoint methods, and more for ODEs, SDEs, DDEs, DAEs, etc.
Rumale is a machine learning library in Ruby
DeepTab is a Python package that simplifies tabular deep learning by providing a suite of models for regression, classification, and distributional regression tasks. It includes models such as Mambular, TabM, FT-Transformer, TabulaRNN, TabTransformer, and tabular ResNets.
Deep learning framework for MRI reconstruction
OpenNN - Open Neural Networks Library
DAMO-ConvAI: The official repository which contains the codebase for Alibaba DAMO Conversational AI.
24,540 repositories in the index in total.