Diferenciación automática de Fortran con LFortran y Enzyme

Differentiable Fortran with LFortran and Enzyme

Este artículo explora cómo obtener gradientes exactos de código Fortran heredado sin reescribirlo, combinando LFortran, LLVM y Enzyme a nivel de IR. El equipo de Pasteur Labs demuestra el proceso con un solucionador de calor 2D en Fortran 90, detallando el pipeline de compilación de seis pasos, los desafíos encontrados (como la necesidad de evitar optimizaciones agresivas antes del paso AD) y la integración del resultado como primitiva personalizada en JAX. El resultado: un motor de física diferenciable que permite retropropagación a través de simulaciones existentes, abriendo la puerta a problemas inversos complejos.