refactor: consolidate backend logic into array_api.py with generic implementations#5202

njzjz merged 8 commits into

masterdeepmodeling/deepmd-kit:masterfrom

copilot/refactor-jax-backend-logicdeepmodeling/deepmd-kit:copilot/refactor-jax-backend-logicCopy head branch name to clipboard