Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
Nov 6, 2025 - Python
#
gpu
Here are 983 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
Nov 6, 2025 - 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
Nov 5, 2025 - Python
Open deep learning compiler stack for cpu, gpu and specialized accelerators
javascript machine-learning performance deep-learning metal compiler gpu vulkan opencl tensor spirv rocm tvm
-
Updated
Nov 6, 2025 - Python
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
-
Updated
Nov 6, 2025 - Python
Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, 17+ clouds, or on-prem).
data-science machine-learning deep-learning gpu job-scheduler cloud-management spot-instances cloud-computing job-queue hyperparameter-tuning cost-optimization distributed-training multicloud ml-infrastructure tpu cost-management finops ml-platform llm-serving llm-training
-
Updated
Nov 6, 2025 - 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
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
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
Oct 14, 2025 - 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 27, 2025 - 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
A Python framework for accelerated simulation, data generation and spatial computing.
-
Updated
Nov 6, 2025 - 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
Mar 21, 2025 - 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
Nov 4, 2025 - Python
📊 A simple command-line utility for querying and monitoring GPU status
-
Updated
Apr 13, 2025 - Python
On-device AI across mobile, embedded and edge for PyTorch
-
Updated
Nov 6, 2025 - Python
Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
-
Updated
Jul 28, 2025 - Python
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.
-
Updated
Nov 6, 2025 - Python
Improve this page
Add a description, image, and links to the gpu topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the gpu topic, visit your repo's landing page and select "manage topics."