ModuleDict ========== Holds submodules in a dictionary. **Abstract Signature:** ``ModuleDict(modules: Dict[str, Module])`` .. raw:: html

PyTorch

API: torch.nn.ModuleDict
Strategy: Direct Mapping

JAX (Core)

API:
Strategy: Custom / Partial

Keras

API:
Strategy: Custom / Partial

TensorFlow

API:
Strategy: Custom / Partial

Apple MLX

API:
Strategy: Custom / Partial

Flax NNX

API:
Strategy: Custom / Partial