Add jax/torch/gguf(fp16) for checkpoints 10k/20k/30k (params only, no train_state)

initial commit