Worked on EnzymeAD’s Enzyme and Enzyme-JAX repositories to expand automatic differentiation capabilities and improve test reliability. Developed forward-mode differentiation support for exponential operations by implementing derivative calculations for LLVM’s ExpOp in MLIR, using C++ and TableGen to define new operations and corresponding tests. Addressed stability in the MLIR pass pipeline by correcting a pass registration typo, ensuring proper invocation of compiler passes. Enhanced test robustness by introducing Type-Based Alias Analysis (TBAA) tags to LLVM load and store operations in the PTX kernel path, which improved memory access analysis and reduced CI noise. Focused on low-level optimization and compiler development.