tests/studio: tighten MLX smoke gates (loss + round-trip, _on_step grad_norm)#5537

Merged

danielhanchen merged 9 commits into

mainfrom

studio-mlx-smoke-gates

May 18, 2026

Commits on May 18, 2026

tests/studio: accept new grad_norm arg in MLX smoke _on_step callback
danielhanchen
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
danielhanchen
committed
tests/studio: pin max_grad_value=0 in MLX smoke so max_grad_norm=1.0 wins
danielhanchen
committed
tests/studio: clarify why MLX smoke pins max_grad_value=0
danielhanchen
committed
tests/studio: replace fragile substring gate with loss + round-trip gates
danielhanchen
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
danielhanchen
committed
tests/studio: gate MLX reload on training-row loss, not greedy text
danielhanchen
committed
tests/studio: align MLX smoke with elementwise-clip + 30-step gates
danielhanchen
committed
tests/studio: hard gate via teacher-forced completion loss
danielhanchen
committed