API Reference# Model Main Model Heads Embeddings Layers Core Layers Convolutions Attention Training Training Utilities Losses Metrics Data Transforms