Skip to content

Add reverse-mode differentiable jax.lax.while_loop to jax_utils.

cfb5920
Select commit
Loading
Failed to load commit list.
Merged

Add reverse-mode differentiable jax.lax.while_loop to jax_utils. #1437

Add reverse-mode differentiable jax.lax.while_loop to jax_utils.
cfb5920
Select commit
Loading
Failed to load commit list.