artisinal-lm

feat: speciy desired KL divergence instead of number of steps

fix: loading checkpoint trained on GPU on CPU-only pytorch requires device flag

fix: jaxtyping tensor definition string like this makes linter happy

feat: adds kv cache to attention layers

feat: finalizes/fixes GRPO training loop, calculates per-token logprob values