jax

JAX vmap: понятная векторизация без циклов на Python

Oct. 21, 02:17

JAX vmap и float32 на GPU: дрейф при изменении батча

Oct. 18, 01:31

JAX и jit: TracerArrayConversionError и как исправить

Oct. 17, 08:16

JAX: scan, pytrees и JIT-кеш — как избежать перекомпиляции

Oct. 3, 07:16

JAX lax.scan: lambda, def или carry — что выбрать и почему

Oct. 2, 21:17

Расписания Optax в JAX/Flax: управление гиперпараметрами

Sep. 21, 20:01

1