PyTorch
API: torch.complex
Strategy: Direct Mapping
JAX (Core)
API: —
Strategy: Macro '{real} + 1j * {imag}'
NumPy
API: numpy.f2py.symbolic.Op.COMPLEX
Strategy: Macro '{real} + 1j * {imag}'
Keras
API: —
Strategy: Macro 'keras.ops.add({real}, keras.ops.multiply(1j, {imag}))'
TensorFlow
API: tf.complex
Strategy: Direct Mapping
Apple MLX
API: complex
Strategy: Direct Mapping
Flax NNX
API: jax.numpy.complex_
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.complex_
Strategy: Direct Mapping