为何pop()方法需同时返回栈顶元素与更新后的栈？

• 贴合函数式编程的不可变特性：Jax走的是函数式编程路线，核心规矩就是数据不能原地修改。普通Python栈的pop会直接改原栈的状态，但在Jax里不行——所有数据结构都是不可变的，动不了原对象。所以pop只能返回一个新的栈实例，顺便把弹出的元素一起给你，这样既拿到了想要的元素，又能得到更新后的栈状态。

• 适配Jax的编译与自动微分机制：Jax的jit编译、自动微分这些核心功能，要求函数必须是纯函数——输入一样输出就一样，不能有偷偷修改外部状态的副作用。要是pop直接改原栈，就违反了纯函数的要求，轻则jit编译报错，重则自动微分过程中状态乱套。返回新栈的方式能保证函数没副作用，完美适配Jax的这些特性。

• 避免状态混乱，提升并行安全性：在多线程、并行计算或者Jax的向量化操作场景里，共享可变状态很容易搞出竞态条件，或者让状态变得不可预测。用不可变栈+返回新实例的方式，每个操作都是独立的，状态明明白白，不用额外加锁或者同步，代码更靠谱。

• 方便链式操作：这种设计能让代码更简洁，比如你可以直接把返回的新栈传给下一个操作，不用单独维护栈的状态变量。像elem, new_stack = stack.pop()之后，直接拿new_stack继续压栈、弹出，逻辑连贯不拖沓。

内容的提问来源于stack exchange，提问作者vtm11