jax.lax.stop_gradient should be not useless in an example#2635
Open
faucct wants to merge 1 commit intod2l-ai:masterfrom
Open
jax.lax.stop_gradient should be not useless in an example#2635faucct wants to merge 1 commit intod2l-ai:masterfrom
faucct wants to merge 1 commit intod2l-ai:masterfrom
