Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
May 25, 2024 - Python
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 Unified AI Framework
Open deep learning compiler stack for cpu, gpu and specialized accelerators
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
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.
Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, Baichuan, Mixtral, Gemma, etc.) on Intel CPU and GPU (e.g., local PC with iGPU, discrete GPU such as Arc, Flex and Max). A PyTorch LLM library that seamlessly integrates with llama.cpp, Ollama, HuggingFace, LangChain, LlamaIndex, DeepSpeed, vLLM, FastChat, etc.
A flexible framework of neural networks for deep learning
SkyPilot: Run LLMs, AI, and Batch jobs on any cloud. Get maximum savings, highest GPU availability, and managed execution—all with a simple interface.
High-performance TensorFlow library for quantitative finance.
An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.
📊 A simple command-line utility for querying and monitoring GPU status
Time series forecasting with PyTorch
Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
A Pythonic framework to simplify AI service building
Add a description, image, and links to the gpu topic page so that developers can more easily learn about it.
To associate your repository with the gpu topic, visit your repo's landing page and select "manage topics."