You can deploy this model and start earning money today!

codezakh/
gpu-forecasters-gpt-oss-20b-correctness-brier

Name: gpu-forecasters-gpt-oss-20b-correctness-brier
Availability: InStock

safetensors

2605.31464

Views

No views yet

Companion artifact for GPU Forecasters: Language Models as Selective Surrogates for Kernel Runtime Optimization. Code: codezakh/gpu-surrogates.

LoRA adapter for openai/gpt-oss-20b fine-tuned with the correctness + Brier reward to forecast kernel speedups.

Loading

python

from peft import AutoPeftModelForCausalLM
model = AutoPeftModelForCausalLM.from_pretrained("codezakh/gpu-forecasters-gpt-oss-20b-correctness-brier")

Training data

codezakh/gpu-forecasters-rl-training-pool.

Reproducing

See runbook/02_train_surrogate.py in the paper repo (codezakh/gpu-surrogates).

Citation

bibtex

@article{khan2026gpuforecasters,
  title={GPU Forecasters: Language Models as Selective Surrogates for Kernel Runtime Optimization},
  author={Khan, Zaid and Chen, Justin Chih-Yao and Cho, Jaemin and Stengel-Eskin, Elias and Bansal, Mohit},
  journal={arXiv preprint arXiv:2605.31464},
  year={2026}
}