GELU

Auto-generated from jax_code_defs

PyTorch

API: torch.nn.GELU
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.nn.gelu
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.layers.Activation
Strategy: Inline Lambda
Official Docs ↗

TensorFlow

API: keras.ops.gelu
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.nn.GELU
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.gelu
Strategy: Direct Mapping
Official Docs ↗