·7 min read
Deep Learning with JAX
Accelerate machine learning models with JAX: master automatic differentiation, jit compilation, vmap vectorization, and TPU/GPU tensor parallel execution.
Technical Writings
Stories, tutorials, and deep dives spanning across diverse worlds.
Top Topics
3 results
·10 min read
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.