PyTorch
API: str
Strategy: Direct Mapping
Keras
API: str
Strategy: Direct Mapping
TensorFlow
API: str
Strategy: Direct Mapping
Apple MLX
API: str
Strategy: Direct Mapping
Flax NNX
API: flax.nnx.PARTITION_NAME
Strategy: Direct Mapping
PaxML / Praxis
API: str
Strategy: Direct Mapping