fix TracerBoolConversionError by using lax.cond instead of if/else, for jax>=0.8.2 bug#188
Closed
callumtilbury wants to merge 1 commit intopatrick-kidger:devfrom
Closed
fix TracerBoolConversionError by using lax.cond instead of if/else, for jax>=0.8.2 bug#188callumtilbury wants to merge 1 commit intopatrick-kidger:devfrom
callumtilbury wants to merge 1 commit intopatrick-kidger:devfrom
