uv run train_gpt_medium.py