Add support for while_loop capture

[obliviateandsurrender]

Context: This add support for capturing while_loop into plxpr.

Description of the Change: The function's behaviour resembles the one for for_loop, and works on capturing both cond_fn and body_fn and executing both of them at runtime.

Benefits: while_loop will enjoy plxpr when capture is enabled

Possible Drawbacks: N/A

Related GitHub Issues: [sc-66773]

[github-actions]

Hello. You may have forgotten to update the changelog!
Please edit doc/releases/changelog-dev.md with:

• A one-to-two sentence description of the change. You may include a small working example for new features.
• A link back to this PR.
• Your name (or GitHub username) in the contributors section.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.66%. Comparing base (4934593) to head (1c68675).
Report is 1 commits behind head on master.

Additional details and impacted files

@@           Coverage Diff            @@
##           master    #6064    +/-   ##
========================================
  Coverage   99.66%   99.66%            
========================================
  Files         430      430            
  Lines       41786    41517   -269     
========================================
- Hits        41644    41377   -267     
+ Misses        142      140     -2     

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

[mudit2812]

Nice!

[albi3ro]

Thanks for putting this together 🎉