Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
Oct 21, 2026 - Python
#
gpu
Here are 3,097 public repositories matching this topic...
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
machine-learning compression deep-learning gpu inference pytorch zero data-parallelism model-parallelism mixture-of-experts pipeline-parallelism billion-parameters trillion-parameters
-
Updated
Oct 20, 2026 - Python
Open Machine Learning Compiler Framework
javascript machine-learning performance deep-learning metal compiler gpu vulkan opencl tensor spirv rocm tvm
-
Updated
Oct 20, 2026 - Python
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
python cpu profiler gpu performance-analysis memory-allocation profiling cpu-profiling memory-consumption gpu-programming python-profilers scalene profiles-memory performance-cpu
-
Updated
Oct 20, 2026 - Python
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
-
Updated
Oct 20, 2026 - Python
The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.
machine-learning deep-learning gpu job-scheduler slurm cloud-management spot-instances cloud-computing job-queue hyperparameter-tuning cost-optimization distributed-training multicloud ml-infrastructure tpu mlops ml-platform llm-serving llm-training
-
Updated
Oct 21, 2026 - Python
A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
python machine-learning algorithm video gpu detection prediction python3 artificial-intelligence artificial-neural-networks image-recognition densenet object-detection squeezenet inceptionv3 offline-capable image-prediction imageai ai-practice-recommendations
-
Updated
Aug 3, 2024 - Python
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.
-
Updated
Jan 28, 2026 - Python
Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
tracking skeleton gpu realtime pytorch keypoints human-pose-estimation human-computer-interaction pose-estimation accurate human-tracking posetracking alphapose human-pose-tracking alpha-pose person-pose-estimation crowdpose full-body whole-body human-joints
-
Updated
May 13, 2024 - Python
An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.
console monitoring gpu grafana cuda prometheus nvidia prometheus-exporter curses nvml top command-line-tool htop grafana-dashboard nvidia-smi monitoring-tool process-monitoring gpu-monitoring resource-monitor
-
Updated
Oct 17, 2026 - Python
A Python framework for GPU-accelerated simulation, robotics, and machine learning.
-
Updated
Oct 20, 2026 - Python
《深入理解 AI Infra：量化分析与系统设计》（李博杰 著）开源书稿：从硬件约束和模型架构出发，量化推导 LLM 推理与训练系统设计。含全书正文、PDF、配套计算工具与实验
distributed-systems performance-engineering book gpu accelerator mixture-of-experts kv-cache ai-infra llm llm-serving llm-training llm-inference deepseek ai-infrastructure datacenter-network
-
Updated
Oct 1, 2026 - Python
A flexible framework of neural networks for deep learning
python machine-learning deep-learning neural-network chainer gpu numpy cuda neural-networks cudnn cupy
-
Updated
Aug 28, 2023 - Python
High-performance TensorFlow library for quantitative finance.
python finance tensorflow gpu high-performance quantlib high-performance-computing gpu-computing quantitative-finance numerical-methods numerical-optimization numerical-integration
-
Updated
Aug 6, 2026 - Python
NVIDIA cuML: GPU-Accelerated Machine Learning
-
Updated
Oct 20, 2026 - Python
On-device AI across mobile, embedded and edge for PyTorch
-
Updated
Oct 20, 2026 - Python
Time series forecasting with PyTorch
python data-science machine-learning ai timeseries deep-learning gpu pandas pytorch artificial-intelligence uncertainty neural-networks forecasting hacktoberfest temporal timeseries-forecasting pytorch-lightning
-
Updated
Oct 20, 2026 - Python
Achieve state of the art inference performance with modern accelerators on Kubernetes
-
Updated
Oct 20, 2026 - Python
Add this topic to your repo
To associate your repository with the gpu topic, visit your repo's landing page and select "manage topics."