pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
GitHub2026-10-11
9 条相关内容
Tensors and Dynamic neural networks in Python with strong GPU acceleration
A unified library of SOTA model optimization techniques like quantization, distillation, pruning, neural architecture search, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM, TensorRT, vLLM, etc. to optimize inference speed.