break gradient and parameters hooks

[kptkin]

Fixes WB-8217

Description

What does the PR do?

Breaks the hooks for gradient and parameters. The hook for gradients seems to work with jit but the hook for parameters is currently can't be scripted. This PR doesn't fix the issue for people that want to track weights, but it help for people that want to track just gradients.

Testing

How was this PR tested?

Checklist

• Name PR "[WB-NNNN][WB-MMMM] Add support for..." similar to entries in CHANGELOG.md
• Include reference to internal ticket "Fixes WB-NNNN" (and github issue "Fixes #NNNN" if applicable)

[codecov]

Codecov Report

Merging #3509 (0663de0) into master (fa8174f) will decrease coverage by 0.37%.
The diff coverage is 95.65%.

@@            Coverage Diff             @@
##           master    #3509      +/-   ##
==========================================
- Coverage   82.70%   82.32%   -0.38%     
==========================================
  Files         256      256              
  Lines       32512    32519       +7     
==========================================
- Hits        26889    26772     -117     
- Misses       5623     5747     +124     

Flag	Coverage Δ
functest	54.23% <84.78%> (-0.78%)	⬇️
unittest	73.52% <93.47%> (+0.08%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
wandb/wandb_torch.py	61.29% <94.11%> (+1.35%)	⬆️
wandb/sdk/wandb_watch.py	81.48% <100.00%> (+6.93%)	⬆️
wandb/integration/prodigy/__init__.py	0.00% <0.00%> (-100.00%)	⬇️
wandb/integration/prodigy/prodigy.py	0.00% <0.00%> (-95.00%)	⬇️
wandb/sdk/data_types/base_types/media.py	91.36% <0.00%> (-1.44%)	⬇️
wandb/sdk/wandb_run.py	91.18% <0.00%> (-0.19%)	⬇️
wandb/sdk/lib/git.py	79.01% <0.00%> (ø)
wandb/sdk/lib/sock_client.py	92.08% <0.00%> (ø)
wandb/sdk/wandb_init.py	85.90% <0.00%> (+0.18%)	⬆️
wandb/sdk/wandb_setup.py	89.00% <0.00%> (+0.50%)	⬆️
... and 2 more

[dmitryduev]

LGTM!