Czy wiesz, że tinygrad obsługuje vmap? W wielu aspektach nasz sposób realizacji jest czystszy niż w JAX, chociaż składnia nie jest najlepsza. Ktoś chce poprawić składnię i napisać samouczek?