Pallas：JAX 内核语言完整指南 on Pengcheng's Bloghttps://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/Recent content in Pallas：JAX 内核语言完整指南 on Pengcheng's BlogHugo -- gohugo.iozh-CN© 2026 PengchengSun, 19 Apr 2026 00:00:00 +0000Pallas 快速入门https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_quickstart_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_quickstart_cn/Grid 和 BlockSpechttps://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_grid_blockspec_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_grid_blockspec_cn/软件流水线https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_pipelining_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_pipelining_cn/使用 Pallas 编写 TPU 内核https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_details_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_details_cn/TPU 流水线https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_pipelining_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_pipelining_cn/矩阵乘法https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_matmul_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_matmul_cn/标量预取与块稀疏计算https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_sparse_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_sparse_cn/TPU 上的 Pallas 分布式计算https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_distributed_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_distributed_cn/Pallas 核心特定编程https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_coremap_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_coremap_cn/SparseCore 内核编写https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_sparsecore_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_sparsecore_cn/伪随机数生成https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_prng_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_tpu_prng_cn/使用 Pallas 编写 Mosaic GPU 内核https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_gpu_reference_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_gpu_reference_cn/Mosaic GPU 流水线https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_gpu_pipelining_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_gpu_pipelining_cn/Blackwell 高性能矩阵乘法https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_gpu_blackwell_matmul_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_gpu_blackwell_matmul_cn/集合矩阵乘法https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_gpu_collective_matmul_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_gpu_collective_matmul_cn/Pallas 设计https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_design_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_design_cn/Pallas 异步操作https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_design_async_cn/Sun, 19 Apr 2026 00:00:00 +0000https://pengchengneo.github.io/posts/pallas-jax-kernel-language-cn/pallas_design_async_cn/