feat (tf/pt): add atomic weights to tensor loss (#4466) #3499

Job	Run time
determine-arm64-runner	1s
Build source distribution	11s
Build wheels for cp311-manylinux_x86_64	6m 6s
Build wheels for cp311-manylinux_x86_64	5m 55s
Build wheels for cp311-macosx_x86_64	5m 59s
Build wheels for cp311-macosx_arm64	3m 16s
Build wheels for cp311-win_amd64	5m 4s
Build wheels for cp310-manylinux_aarch64	12m 30s
build_docker	1s
build_pypi_index	1s
upload_pypi	1s
deploy_pypi_index	1s
Pass testing build wheels	1s
	39m 7s

Provide feedback