`lax.select` versus `(1-x)*y + x*z`
RobertTLange opened this issue · comments
Robert Tjarko Lange commented
As of right now I am doing a lot masked sums to avoid booleans while jitting. This can probably done a lot more elegantly with lax.select
as in the podracer Anakin example. Go through all environments and clean up!