simonts/genre2-rlm-ntl-was-kbss-partial

GenRe2 RLM NTL-WAS KBSS Partial Checkpoint

This repository is a partial RLM checkpoint release candidate. It is not the final Table 2 RLM GenRe2 artifact.

Public Hub repo used for this release:

simonts/genre2-rlm-ntl-was-kbss-partial

Source

Local source path used for this release:

/home/trx/rlm-code/ICLR26/results_exp_rlm_ntl_was/kbss/checkpoints/checkpoint_best

Expected uploaded files by default:

• latest
• zero_to_fp32.py
• pytorch_model/*model_states.pt

Optimizer shards and RNG state are intentionally not uploaded by default.

Result Status

The local single-seed result found for this checkpoint is:

Local result	Spearman	Paper target
results_exp_rlm_ntl_was/kbss/results_42.json	0.5334	NTL-WAS Triton 0.539 +/- 0.010

The rank correlation is within the paper range for NTL-WAS Triton, but this is only a partial single-task/single-seed checkpoint. The raw RMSE/R2 values in the local result are poor, and the full five-seed Table 2 artifact was not found.

Important Limitation

Do not present this repository as the final RLM ReMax or GenRe2 weight release. The local search did not find complete RLM ReMax/GenRe2 weights or five-seed result JSONs for the paper Table 2 APPS/Triton/NAS experiments.

Conversion

If a consolidated fp32 checkpoint is needed, use:

python zero_to_fp32.py . /tmp/rlm_ntl_was_kbss_fp32.pt

Notes

• The model card license is MIT to match the upstream public base checkpoint metadata for akhauriyash/RLM-GemmaS-Code-v0.
• This repository is published only because it is clearly labeled as partial.