Fix checkpoint conversion script

[shuningjin]

Description

Fix imports after the file location change in #2601.

Scripts are currently broken for deepseek, qwen3-moe, and llama_mistral_mixtral_orbax_to_hf.

Tests

python3 -m MaxText.utils.ckpt_scripts.convert_deepseek_family_ckpt \
--base_model_path /home/shuningjin/deepseek2-16b/hf-16b \
--maxtext_model_path gs://runner-maxtext-logs/test-ds2-conv \
--model_size deepseek2-16b

https://paste.googleplex.com/6609477168136192

Checklist

Before submitting this PR, please make sure (put X in square brackets):

• I have performed a self-review of my code. For an optional AI review, add the gemini-review label.
• I have necessary comments in my code, particularly in hard-to-understand areas.
• I have run end-to-end tests tests and provided workload links above if applicable.
• I have made or will make corresponding changes to the doc if needed, including adding new documentation pages to the relevant Table of Contents (toctree directive) as explained in our documentation.