Decoradores JIT de Python: Tres estrategias de implementación

Este artículo profundiza en el popular patrón de decorador JIT en Python, particularmente su uso en las bibliotecas JAX y Triton. El autor implementa tres decoradores JIT desde cero utilizando un ejemplo simplificado: basado en AST, basado en bytecode y basado en rastreo. El enfoque basado en AST manipula directamente el Árbol de Sintaxis Abstracto; el enfoque basado en bytecode aprovecha el intérprete de bytecode de Python; y el enfoque basado en rastreo construye un IR de expresión rastreando la ejecución de la función en tiempo de ejecución. El artículo detalla las ventajas y desventajas de cada enfoque y utiliza JAX y Numba como ejemplos para ilustrar sus estrategias en aplicaciones del mundo real.