JAX — это мощный фреймворк от Google для высокопроизводительных вычислений и машинного обучения, который завоевал любовь исследователей своей скоростью и элегантным функциональным подходом. Однако его...