PyTorch
API: torch.TYPE_CHECKING
Strategy: Direct Mapping
JAX (Core)
API: typing.TYPE_CHECKING
Strategy: Direct Mapping
NumPy
API: typing.TYPE_CHECKING
Strategy: Direct Mapping
Keras
API: typing.TYPE_CHECKING
Strategy: Direct Mapping
TensorFlow
API: typing.TYPE_CHECKING
Strategy: Direct Mapping
Apple MLX
API: typing.TYPE_CHECKING
Strategy: Direct Mapping
Flax NNX
API: typing.TYPE_CHECKING
Strategy: Direct Mapping
PaxML / Praxis
API: typing.TYPE_CHECKING
Strategy: Direct Mapping