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