jax
JAX: scan vs Python for под jit — как выбрать быстрее
Dec. 30, 03:02
JAX и списки Python: как dtype срывает оптимизацию RNN
Dec. 28, 15:02
JAX: эффективная сборка изображений из патчей без циклов
Dec. 19, 09:01
JAX vmap: понятная векторизация без циклов на Python
Oct. 21, 02:17
JAX vmap и float32 на GPU: дрейф при изменении батча
Oct. 18, 01:31
JAX и jit: TracerArrayConversionError и как исправить
Oct. 17, 08:16
Расписания Optax в JAX/Flax: управление гиперпараметрами
Sep. 21, 20:01