Tensors and Dynamic neural networks in Python with strong GPU acceleration
- neural-network
- autograd
- gpu
- numpy
47,133 repositories indexed
1–15 of 15 · Most stars
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.
If you live in the terminal, kitty is made for you! Cross-platform, fast, feature-rich, GPU based.
Open Machine Learning Compiler Framework
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
NumPy & SciPy for GPU
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.
《深入理解 AI Infra:量化分析与系统设计》(李博杰 著)开源书稿:从硬件约束和模型架构出发,量化推导 LLM 推理与训练系统设计。含全书正文、PDF、配套计算工具与实验
High-performance TensorFlow library for quantitative finance.
NVIDIA cuML: GPU-Accelerated Machine Learning
PyGraphistry is a Python library to quickly load, shape, embed, and explore big graphs with the GPU-accelerated Graphistry visual graph analyzer
The animation engine for agents: blazingly fast math and 3D videos with Manim's Python API, on a Rust GPU core.
DNABERT: pre-trained Bidirectional Encoder Representations from Transformers model for DNA-language in genome
🧱 Experimental library for building inference engines