Google JAX

Google JAX Google JAX — платформа машинного обучения для преобразования числовых функций в Python.  JAX объединяет модифицированную версию autograd и […]

Google JAX

  • Google JAX — платформа машинного обучения для преобразования числовых функций в Python. 
  • JAX объединяет модифицированную версию autograd и XLA от TensorFlow. 
  • Он разработан для соответствия структуре и рабочему процессу NumPy и работает с различными фреймворками. 
  • Основные функции JAX включают градиентную автоматическую дифференциацию, jit-компиляцию, виртуальную карту и pmap. 
  • Несколько библиотек Python используют JAX в качестве серверной части, включая Flax, Equinox, Optax, RLax и jraph. 
  • JAX также используется в некоторых библиотеках R для ускорения линейной алгебры и разработки агентов обучения с подкреплением. 

Полный текст статьи:

Google JAX — Википедия

Оставьте комментарий

Прокрутить вверх