VitExitLayers ============= Exit block of ViT. **Abstract Signature:** ``VitExitLayers(hidden_dim: int, output_dim: int)`` .. raw:: html

PyTorch

API:
Strategy: Custom / Partial

PaxML / Praxis

API: praxis.layers.VitExitLayers
Strategy: Direct Mapping