GeluApprox

An approximation to Gaussian Error Linear Unit (Tanh based).

Abstract Signature:

GeluApprox(x: Tensor)

PyTorchJAX (Core)Apple MLX

PyTorch

API: torch.nn.functional.gelu

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.nn.gelu

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.gelu_approx

Strategy: Direct Mapping

Official Docs ↗