offer a naive unoptimized attention w/ stop graddable queries, keys, …

…values, and fix the self reasoning transformer to only stop grad keys and values for the reasoning tokens