Тег

#pytorch

Все топики с тегом #pytorch.

Sonata

Higgsfield — фреймворк для распределённой тренировки больших моделей

Higgsfield — это открытый, отказоустойчивый и высокомасштабируемый инструмент оркестрации GPU, а также фреймворк для машинного обучения, созданный для тренировки моделей с миллиардами и триллионами параметров, включая большие языковые модели (LLM).

Higgsfield выполняет роль менеджера GPU-задач и объединяет несколько ключевых функций: распределение эксклюзивного и неэксклюзивного доступа к вычислительным узлам, поддержку ZeRO-3 DeepSpeed API и Fully Sharded Data Parallel API от PyTorch для эффективного шардинга моделей с триллионом параметров, а также запуск, выполнение и мониторинг тренировки больших нейросетей.

DS

PyTorch: библиотека для глубокого обучения на GPU

PyTorch — Python-библиотека для тензорных вычислений с GPU-ускорением, аналогично NumPy. Она предоставляет ленточную систему автоматического дифференцирования (autograd) для построения глубоких нейронных сетей.

Ключевая особенность PyTorch — динамические нейронные сети на основе tape-based autograd. Это позволяет менять архитектуру на ходу без перестроения графа. Библиотека глубоко интегрирована с Python, упрощая отладку и написание кода.