[Jax reference](https://docs.jax.dev/en/latest/_autosummary/jax.lax.one_minus_square.html). (I will submit a PR if this is wanted.)
Activity
NeilGirdhar commentedon Oct 22, 2026
This would allow you write nice Array API code while still accessing the Lax function for Jax's case.