fix(dpmodel/pt/pd/jax): pass trainable to layer & support JAX trainable & support TF tensor fitting trainable#4793

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

njzjz:trainablenjzjz/deepmd-kit:trainableCopy head branch name to clipboard