You are working on the blackjax-devs/blackjax repository. The source code is checked out at /testbed at…

You are working on the blackjax-devs/blackjax repository. The source code is checked out at /testbed at…: a task in SWE-Lego-RL-2699 (Harbor dataset). This would simplify the code greatly, provide a small performance improvement for those who use the lower-level directly, and make the internals…