NVIDIA mở mã nguồn Model Optimizer giúp tối ưu và nén mô hình AI

NVIDIA đã chính thức mở mã nguồn dự án NVIDIA Model Optimizer (ModelOpt), một thư viện hợp nhất chứa các kỹ thuật tối ưu hóa mô hình AI tiên tiến nhất hiện nay. Dự án cho phép các nhà phát triển nén mô hình hiệu quả nhằm tăng tốc độ suy luận khi triển khai trên các hạ tầng thực tế.

ModelOpt tích hợp đầy đủ các phương pháp hàng đầu như lượng hóa (quantization), chắt lọc tri thức (distillation), cắt tỉa (pruning), tìm kiếm kiến trúc mạng (NAS) và speculative decoding. Thư viện hỗ trợ đầu vào từ PyTorch, Hugging Face và ONNX, đồng thời liên kết chặt chẽ với NVIDIA Megatron-LM và Hugging Face Accelerate để phục vụ cho các quá trình tối ưu trong lúc huấn luyện.

Các checkpoint mô hình sau khi tối ưu có thể xuất trực tiếp sang các framework suy luận phổ biến như TensorRT-LLM, vLLM, TensorRT và SGLang. Đặc biệt, NVIDIA cũng bổ sung khả năng tích hợp kỹ năng agent cho các trợ lý như Claude và Codex, giúp nhà phát triển thao tác trực tiếp trong không gian làm việc.

Đây là tin vui lớn cho các lập trình viên AI Việt Nam, giúp việc tối ưu dung lượng và tăng tốc suy luận các mô hình LLM lớn trở nên dễ dàng hơn trên hạ tầng GPU NVIDIA.

Nguồn: GitHub Trending — Biên dịch & tổng hợp: danhbaai.com

Danh Bạ AI
Logo
Register New Account