fixed jnp.clip deprecation

[chiamp]

fixed jnp.clip deprecation warning

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 60.36%. Comparing base (2c7d7cd) to head (fb5515c).
Report is 7 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3905      +/-   ##
==========================================
- Coverage   60.43%   60.36%   -0.07%     
==========================================
  Files         105      105              
  Lines       13263    13276      +13     
==========================================
- Hits         8015     8014       -1     
- Misses       5248     5262      +14     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.