Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
Apr 28, 2025 - Python
#
gpu
Here are 862 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
Apr 28, 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
Apr 18, 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
Apr 28, 2025 - Python
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
-
Updated
Apr 28, 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
SkyPilot: Run AI and batch jobs on any infra (Kubernetes or 16+ clouds). Get unified execution, cost savings, and high GPU availability via a simple interface.
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
Apr 28, 2025 - 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
Apr 28, 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
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
Apr 25, 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 uncertainty neural-networks forecasting temporal artifical-intelligense timeseries-forecasting pytorch-lightning
-
Updated
Apr 18, 2025 - Python
📊 A simple command-line utility for querying and monitoring GPU status
-
Updated
Apr 13, 2025 - Python
Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
-
Updated
Apr 22, 2025 - Python
On-device AI across mobile, embedded and edge for PyTorch
-
Updated
Apr 28, 2025 - Python
A Pythonic framework to simplify AI service building
-
Updated
Apr 17, 2025 - Python
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.
-
Updated
Apr 25, 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."