bias_correctionΒΆ

Auto-generated from jax_code_defs

JAX (Core)

API: optax.tree.bias_correction
Strategy: Direct Mapping

Apple MLX

API: mlx.optimizers.optimizers.Adam.bias_correction
Strategy: Direct Mapping

Flax NNX

API: optax.tree.bias_correction
Strategy: Direct Mapping