Python 的 JIT 装饰器：三种实现策略详解

本文深入探讨了 Python 中流行的 JIT 装饰器模式，特别是 JAX 和 Triton 库中的应用。作者通过一个简化的示例，从零开始实现了三种 JIT 装饰器：基于 AST 的、基于字节码的和基于追踪的。基于 AST 的方法直接处理抽象语法树；基于字节码的方法利用 Python 的字节码解释器；基于追踪的方法则通过运行时追踪函数执行来构建表达式 IR。文章详细比较了这三种方法的优缺点，并以 JAX 和 Numba 为例，阐述了它们在实际应用中的策略。