Skip to content

Fix the gradient contains nan bug

Compare
Choose a tag to compare
@Szkered Szkered released this 20 Jul 05:36
· 10 commits to main since this release
773cdfe

lax.select generates a nan in the gradient from the inactive branch, while lax.cond doesn't