Add complex type mappings for f32 and f64 (#2026)

Author: ptiede

src/enzyme_ad/jax/primitives.py

@@ -953,6 +953,8 @@ def maketup(ty):
         "i64": "int64_t",
         "ui32": "uint32_t",
         "ui64": "uint64_t",
+        "complex<f32>": "std::complex<float>",
+        "complex<f64>": "std::complex<double>",
     }[tystr]
     return (tystr, ty.shape)