Improve Adam control_state-related #5261

Job	Run time
Dependency	4s
Shellcheck	3s
Prepare	56s
Ruff check and force	19s
Check commit	37s
Create pytest cache for Python 3.13.0	15s
Create pytest cache for Python 3.12	10s
Run pytest using Python 3.13.0	21s
Run pytest using Python 3.12	15s
Process test coverage	20s
Run mypy	18s
Process test complexity	15s
Build and publish Python 🐍 distributions 📦 to TestPyPI	35s
	4m 28s

Provide feedback