Библиотека для оптимизации глубокого обучения, упрощающая и ускоряющая распределённое обучение и инференс больших моделей.
ServiceNow/Fast-LLM
Open-source библиотека для обучения больших языковых моделей с 3D-параллелизмом, оптимизацией памяти ZeRO и интеграцией с Hugging Face на базе PyTorch и Triton.
- Звёзды
- 337
- Форки
- 53
- Открытые issues
- 56
- Последний коммит
- 6 дней назад
Что это?
Fast-LLM — open-source библиотека для обучения больших языковых моделей на базе PyTorch и Triton. Она подходит для моделей примерно от 1B параметров и вплоть до кластеров, обучающих модели с 70B+ параметров, и работает как на одном GPU, так и на многонодовых кластерах под управлением Slurm или Kubernetes с KubeFlow.
Библиотека поддерживает распределённое обучение с параллелизмом по данным, тензорам и конвейеру, параллелизмом по длине последовательности, а также ZeRO-1, ZeRO-2 и ZeRO-3 для эффективного использования памяти. Модели настраиваются через YAML-файлы и запускаются из командной строки. Также в библиотеку входит реализация Mixture-of-Experts без отбрасывания токенов и интеграция с Hugging Face Transformers.
Проект распространяется по лицензии Apache 2.0, а для быстрого развёртывания публикуются готовые Docker-образы. В документации и примерах описана конфигурация обучения Mistral-7B, используемая в README.
Ключевые возможности
- 01Распределённое обучение с параллелизмом по данным, тензорам и конвейеру (3D-параллелизмом).
- 02Оптимизация памяти ZeRO-1, ZeRO-2 и ZeRO-3.
- 03Реализация Mixture-of-Experts (MoE) без отбрасывания токенов.
- 04Обучение с низкой точностью и накопление градиентов.
- 05Конфигурация на основе YAML и интерфейс командной строки.
- 06Интеграция с Hugging Face Transformers.
Сценарии использования
- Предварительное обучение модели в стиле Mistral или GPT на многонодовом GPU-кластере под управлением Slurm.
- Запуск масштабных задач обучения LLM в Kubernetes с PyTorchJob от KubeFlow.
- Обучение собственных MoE-моделей на базе open-source-кода с лицензией Apache 2.0.
Быстрый старт
- 1
Разверните Docker-образ nvcr.io/nvidia/pytorch:25.11-py3 на всех узлах (рекомендуется)
- 2
Установите Fast-LLM на всех узлах с помощью pip
pip install --no-cache-dir "fast-llm[CORE,OPTIONAL] @ git+https://github.com/ServiceNow/Fast-LLM.git" - 3
Отправьте пример скрипта задания Slurm в кластер
sbatch examples/fast-llm.sbat
Что важно знать
- Для обучения в указанном масштабе нужен многузловой GPU-кластер, например из 4 узлов с 8 GPU A100 или H100 на каждом.
- На каждом узле должны быть установлены CUDA 12.1 или новее, а также PyTorch, Triton и Apex.
- Согласно README, некоторые функции, например настраиваемые оптимизаторы и обучающие материалы, всё ещё находятся в разработке.
Темы
Частые вопросы
- Что такое Fast-LLM?
- Open-source библиотека для обучения больших языковых моделей с 3D-параллелизмом, оптимизацией памяти ZeRO и интеграцией с Hugging Face на базе PyTorch и Triton. Fast-LLM — open-source библиотека для обучения больших языковых моделей на базе PyTorch и Triton. Она подходит для моделей примерно от 1B параметров и вплоть до кластеров, обучающих модели с 70B+ параметров, и работает как на одном GPU, так и на многонодовых кластерах под управлением Slurm или Kubernetes с KubeFlow.
- Fast-LLM можно использовать бесплатно? Какая у него лицензия?
- Fast-LLM распространяется с открытым исходным кодом под лицензией Other. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- Fast-LLM всё ещё поддерживается?
- Активно поддерживается. Последний коммит в Fast-LLM — 15 окт. 2026 г..
- Для чего можно использовать Fast-LLM?
- Предварительное обучение модели в стиле Mistral или GPT на многонодовом GPU-кластере под управлением Slurm. Запуск масштабных задач обучения LLM в Kubernetes с PyTorchJob от KubeFlow. Обучение собственных MoE-моделей на базе open-source-кода с лицензией Apache 2.0.
- Как установить Fast-LLM?
- Разверните Docker-образ nvcr.io/nvidia/pytorch:25.11-py3 на всех узлах (рекомендуется) Установите Fast-LLM на всех узлах с помощью pip: pip install --no-cache-dir "fast-llm[CORE,OPTIONAL] @ git+https://github.com/ServiceNow/Fast-LLM.git" Отправьте пример скрипта задания Slurm в кластер: sbatch examples/fast-llm.sbat
- Где скачать Fast-LLM?
- Исходный код и инструкции по установке — на GitHub: https://github.com/ServiceNow/Fast-LLM
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Система на базе PyTorch для распределённого обучения и инференса крупных AI-моделей с поддержкой параллелизма, оптимизации памяти и гетерогенного обучения.
Библиотека PyTorch с архитектурами моделей для обработки изображений, предобученными весами и скриптами обучения, валидации и экспорта для задач компьютерного зрения.
Реализация Vision Transformer (ViT) на PyTorch и множества его вариантов для классификации изображений и других задач компьютерного зрения.
Mamba — архитектура моделирования последовательностей на PyTorch, основанная на селективных моделях пространства состояний и предлагающая альтернативу Transformers с линейной сложностью для языковых и других задач.
Официальный код обучения и эталонная реализация RWKV — языковой архитектуры на основе RNN, которую можно обучать параллельно, как GPT-трансформер.