Python의 JIT 데코레이터: 세 가지 구현 전략

이 글에서는 Python에서 널리 사용되는 JIT 데코레이터 패턴, 특히 JAX 및 Triton 라이브러리에서의 사용에 대해 자세히 설명합니다. 저자는 단순화된 예시를 사용하여 AST 기반, 바이트코드 기반, 추적 기반의 세 가지 JIT 데코레이터를 처음부터 구현합니다. AST 기반 접근 방식은 추상 구문 트리를 직접 조작하고, 바이트코드 기반 접근 방식은 Python의 바이트코드 인터프리터를 활용하며, 추적 기반 접근 방식은 런타임 시 함수 실행을 추적하여 표현식 IR을 구성합니다. 이 글에서는 각 접근 방식의 장단점을 자세히 설명하고, JAX와 Numba를 예시로 들어 실제 애플리케이션에서의 전략을 설명합니다.