Closed RobertTLange closed 2 years ago
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!
lax.select
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!