Словарь
JAX
Открытая библиотека Google для численных вычислений и машинного обучения на Python: она умеет автоматически дифференцировать код, компилировать его через XLA и распараллеливать на ускорителях (GPU и TPU). Используется как альтернатива PyTorch, особенно при обучении крупных моделей на больших кластерах.