[Feature Request] DOS Model consistency test between PT and TF backends

[anyangml]

Summary

Need to investigate the discrepancy between PT and TF backends on DOS Model. The fitting nets are consistent.

Detailed Description

see #4296

Further Information, Files, and Links

No response

[anyangml]

@njzjz Can you add the JAX backend and see if it's consistent with either TF or PT?

[njzjz]

@njzjz Can you add the JAX backend and see if it's consistent with either TF or PT?

The JAX backend reuses codes from the DP backend. What is the status of the DP backend?

[anyangml]

@njzjz Can you add the JAX backend and see if it's consistent with either TF or PT?

The JAX backend reuses codes from the DP backend. What is the status of the DP backend?

DP and PT are consistent.