从 Python 循环展开的问题出发,介绍 lax.scan 的 carry、参数层堆叠和自动微分,并说明如何结合 jax.checkpoint 控制 Transformer 的激活内存。
2026-07-04
JAX 101(八):用 lax.scan 表达重复层与循环状态
JAX 101(七):JAX 中的通用卷积
JAX 101(六):从零拼出迷你 Transformer
JAX 101(五):有状态计算、Ref 与训练循环
JAX 101(四):显式随机数与可复现实验
JAX 101(三):用 PyTree 组织模型参数
JAX 101(二):函数变换、自动微分与批处理
JAX 101(一):数组、jax.numpy 与语言模型的数据表示
使用 DESeq2 和 edgeR 进行 RNA-Seq 差异表达分析
sched_ext 与 scx_lavd 用户态前端解析
Linux CPU 调度与 sched_ext 基础:从任务唤醒到 CPU 执行
使用 libbpf 调试 Futex eBPF 程序
Futex 性能分析
GCC、G++ 与 Windows SDK:C/C++ 混合编译和头文件排查
Rust 语言学习总结
latex报错
内核插桩技术
计算机网络常考八股
配置端口映射踩坑记录
探秘proc和sys下的内存接口
双系统网卡故障排查
Futex 概述
操作系统常问八股3
微服务链路追踪原理浅析