PyTorch
API: torch.logit_
Strategy: Direct Mapping
JAX (Core)
API: jax.scipy.special.logit
Strategy: Direct Mapping
Apple MLX
API: —
Strategy: Macro 'mlx.core.log({input} / (1 - {input}))'
Flax NNX
API: jax.scipy.special.logit
Strategy: Direct Mapping