Jax编程: 踩坑记录

摘要 记录 jax 使用过程中的坑。