This makes running `predict` with the teacher model much faster, when
there are many examples.
Release Notes:
- N/A
---------
Co-authored-by: Ben Kunkle <ben@zed.dev>
ad369ca2b7Max Brunsfeld committed on 1/9/2026, 12:04:40 AM· committed by GitHubparenta842e1d