Главная › Программное обеспечение › JAX
JAX
Программное обеспечение
Обучение
Агенты
Composable function transforms — the framework Gemini is built with.
Перед вами запись о программном обеспечении — инструменте или платформе в AI-стеке: кто её поддерживает и на каком оборудовании она работает.
Фото: Google LLC · Public domain · Wikimedia Commons
JAX treats models as pure functions to be transformed — grad, jit, vmap, pmap — compiling through XLA to TPU pods with exceptional scaling ergonomics. It is Google DeepMind's house framework (Gemini training) and the favorite of much of the research frontier outside PyTorch.
Место в общей экосистеме
Ссылки этой записи с первого взгляда — нажмите на группу, чтобы перейти к ней.
Работает на
Поддерживается
Конкурирует с
Связанные
Источники
Запись актуальна на 2026-01-15 — смотреть Методология.