per_example_layer_norm_clipΒΆ

Auto-generated from jax_code_defs

JAX (Core)

API: optax.per_example_layer_norm_clip
Strategy: Direct Mapping

Flax NNX

API: optax.per_example_layer_norm_clip
Strategy: Direct Mapping