1.pytorch / pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Ranked by Attention Score: stars gained over the last 180 complete days
12 repositories by Attention Score · May 7, 2026 – Nov 2, 2026 (UTC)
Tensors and Dynamic neural networks in Python with strong GPU acceleration
NumPy & SciPy for GPU
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.
Open Machine Learning Compiler Framework
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
High-performance TensorFlow library for quantitative finance.
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
NVIDIA cuML: GPU-Accelerated Machine Learning
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.
PyGraphistry is a Python library to quickly load, shape, embed, and explore big graphs with the GPU-accelerated Graphistry visual graph analyzer
DNABERT: pre-trained Bidirectional Encoder Representations from Transformers model for DNA-language in genome