jax
45a3fe62 - [direct-linearize] add direct linearize rule for cond

Commit

235 days ago

[direct-linearize] add direct linearize rule for cond Co-authored-by: Yash Katariya <yashkatariya@google.com>

References

Author

mattjj

Parents