🚀 ROCm запустил вилку MaxText для оффлайн-разработки
AMD представила структуру своей вилки MaxText с поддержкой ROCm и полностью оффлайн-режима.
Вилка использует двухветковую модель: ветка main зеркалирует upstream, а rocm-main содержит изменения для ROCm и оффлайн-работы. Встроенная папка local_datasets/ включает мини-наборы данных (C4) в форматах ArrayRecord, TFRecord и Parquet для быстрой локальной загрузки. Оффлайн-режим достигается через gcloud_stub.py, pytest-маркеры decoupled и фильтрацию тестов — это исключает зависимости от Google Cloud. Доступны команды вроде pytest -m decoupled для запуска безопасных тестов без сети.
Автоматическая синхронизация с upstream минимизирует конфликты. В планах — аннотированные релизы vX.Y.Z-rocm, дашборды производительности и публикация чекпоинтов для локальных запусков.
