Popular GitHub Repositories for Gpu

Tensors and Dynamic neural networks in Python with strong GPU acceleration

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

The fastai deep learning library

Open deep learning compiler stack for cpu, gpu and specialized accelerators

An open-source, low-code machine learning library in Python

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, DeepSeek, Mixtral, Gemma, Phi, MiniCPM, Qwen-VL, MiniCPM-V, etc.) on Intel XPU (e.g., local PC with iGPU and NPU, discrete GPU such as Arc, Flex and Max); seamlessly integrate with llama.cpp, Ollama, HuggingFace, LangChain, LlamaIndex, vLLM, DeepSpeed, Axolotl, etc.

CUDA Templates for Linear Algebra Subroutines

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

A flexible framework of neural networks for deep learning

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

cuML - RAPIDS Machine Learning Library

Time series forecasting with PyTorch

Deep Learning GPU Training System

PipelineAI

Lightning fast C++/CUDA neural network framework

Fast and flexible AutoML with learning guarantees.

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

On-device AI across mobile, embedded and edge for PyTorch