Technical Writings

Stories, tutorials, and deep dives spanning across diverse worlds.

Search Posts

Học sâu với JAX

Tăng tốc các mô hình học máy với JAX: làm chủ vi phân tự động, biên dịch jit, vector hóa vmap và thực thi song song tensor trên TPU/GPU.