mini-llama / src / optimizer_test.npy