Technologies
JAX
Бібліотека чисельних обчислень нового покоління, що поєднує NumPy‑подібний API з автодиференціюванням і компіляцією через XLA. Вона підтримує векторизацію (vmap), паралелізацію та масштабування на GPU/TPU, що робить її зручною для сучасних досліджень і високопродуктивного тренування. Ми використовуємо JAX для експериментів з новими архітектурами, швидкого