Migrate logits computation and gather to model_runner
#5144
| Job | Run time |
|---|---|
| 36s | |
| 36s |
logits computation and gather to model_runner
#5144
| Job | Run time |
|---|---|
| 36s | |
| 36s |