Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
Wraps XLAβs Select operator.
pred (Any) β
Any
on_true (Any) β
on_false (Any) β