Minqi
Results
2
comments of
Minqi
trafficstars
+1 This would be an incredibly huge time saver!
Hi there! I was wondering what the JAX team's latest thinking is regarding the behavior of lax.cond when batched via vmap. I find myself often running into the design pattern...