jax
dadc68b6 - add experimental lax.optimization_barrier autodiff rules

Commit
313 days ago
add experimental lax.optimization_barrier autodiff rules
Author
Committer
Parents
Loading