fix(jax): setattr case_embd#5104

njzjz merged 8 commits intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom

njzjz:jax-case_embdnjzjz/deepmd-kit:jax-case_embdCopy head branch name to clipboard