Apply function on pytree directly.

blackjax-devs / blackjax

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

https://blackjax-devs.github.io/blackjax/

Apache License 2.0

806 stars 105 forks source link

Closed junpenglao closed 4 months ago

junpenglao commented 4 months ago

Avoiding unnecessary unpacking.

junpenglao commented 4 months ago

@reubenharry Done, thanks!

reubenharry commented 4 months ago

Tests failed!