Подробное описание документа
Сапунов, Г. Глубокое обучение с JAX : руководство / Г. Сапунов ; пер. с англ. А. В. Снастина. — Москва : ДМК Пресс, 2025. — 526 с. — ISBN 978-5-93700-192-4.
Библиотека JAX от Google предоставляет пользователям моделей глубокого обучения средства точной настройки и управления процессами низкого уровня, такими как вычисление градиента, быстрое и эффективное обучение модели и вывод результатов на больших наборах данных. Книга обучает созданию эффективных нейронных сетей с применением JAX. Вы узнаете, как специальные инструменты JAX помогают справиться с проблемами нехватки производительности, характерными для глубокого обучения на больших данных. Примеры реалистичных проектов и листинги исходного кода с подробными комментариями показывают, как парадигма функционального программирования JAX улучшает сочетаемость различных программных компонентов и возможности распараллеливания. Основные темы: использование JAX для числовых расчетов; создание дифференцируемых моделей с использованием примитивов JAX; выполнение распределенных и параллельных вычислений с помощью JAX; создание нейронных сетей с помощью библиотек высокого уровня, таких как Flax. Издание предназначено программистов средней квалификации на языке Python, знакомых с принципами глубокого обучения.