The AGENTS.md file is missing documentation for several new features and we should keep it up to date so AI agents can effectively work with the codebase.

We need to update it to include: OptimizerClient, Container backends, Training options, CustomTrainerContainer, Repository map etc.