JAX: библиотека для матричных вычислений

[vak]

Самому мне редко приходится гонять такие вещи, но штука полезная.

• "JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning"

Вокруг JAX формируется некая экосистема:

• "Flax: A neural network library and ecosystem for JAX designed for flexibility"
• "Equinox is your one-stop JAX library, for everything you need that isn't already in core JAX"