Add new install check

[ko3n1g]

Add another install check like

Megatron-LM/.github/workflows/install-test.yml

Lines 80 to 84 in 0938eb7

	- name: Check imports for megatron.training
	uses: ./FW-CI-templates/.github/actions/check-imports
	with:
	package-name: megatron.training
	python-binary: ${{ env.UV_PROJECT_ENVIRONMENT }}/bin/python

for megatron.inference