PyTorch
API: torch.nn.ModuleDict
Strategy: Direct Mapping
Keras
API: —
Strategy: Custom / Partial
TensorFlow
API: —
Strategy: Custom / Partial
Apple MLX
API: —
Strategy: Custom / Partial
Flax NNX
API: flax.nnx.MultiMetric
Strategy: Direct Mapping
PaxML / Praxis
API: —
Strategy: Custom / Partial