FLOPS-Squared/KeystoneFuse-A-FuserWidth-16-Flax

Upload FlaxLlamaForCausalLM

Checkpoint step 405459

Checkpoint step 400000

Checkpoint step 300000

Checkpoint step 270306

Checkpoint step 200000

Checkpoint step 135153

Checkpoint step 100000

Checkpoint step 32

Checkpoint step 0

Upload tokenizer

initial commit