* flax gpt neo * fix query scaling * update generation test * use flax model for test
Attach a file by drag & drop or click to upload
